Update app.py

app.py

@@ -84,9 +84,9 @@ def preload_data(pdf_urls):
     return index, chunks
 
 index, chunks = preload_data(DEFAULT_PDF_URLS)
-embedding_model = SentenceTransformer("BAAI/bge-large-en")
+embedding_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 accelerator = Accelerator()
-MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
+MODEL_NAME = "google/flan-t5-small"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto", trust_remote_code=True, cache_dir="./my_models")
 model = accelerator.prepare(model)