Skip to content

Instantly share code, notes, and snippets.

@davidmezzetti

davidmezzetti/txtai-qa.py

Last active October 11, 2021 14:41
Show Gist options
  • Save davidmezzetti/a94ffffa565e1ce2c44261a72c3eb297 to your computer and use it in GitHub Desktop.
Save davidmezzetti/a94ffffa565e1ce2c44261a72c3eb297 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
txtai-qa.py
from txtai.embeddings import Embeddings
from txtai.pipeline import Extractor
# Create embeddings model, backed by sentence-transformers & transformers
embeddings = Embeddings({"path": "sentence-transformers/nli-mpnet-base-v2"})
# Create extractor instance
extractor = Extractor(embeddings, "distilbert-base-cased-distilled-squad")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment