Spaces:

random2222
/

trry

Sleeping

random2222 commited on Apr 13

Commit

ecbad47

verified ·

1 Parent(s): 3430630

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def create_qa_system():
         db = FAISS.from_documents(texts, embeddings)
         # Initialize local model with LangChain wrapper
-        model_name = "google/flan-t5-base"
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

         db = FAISS.from_documents(texts, embeddings)
         # Initialize local model with LangChain wrapper
+        model_name = "google/flan-t5-small"
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name)