Update app.py

app.py

@@ -8,8 +8,8 @@ from chromadb.config import Settings
 from transformers import pipeline
 
 # Device setup
-device = 0 if torch.cuda.is_available() else -1
-print(f"Device set to: {'cuda' if device == 0 else 'cpu'}")
+device = -1  # Force CPU use
+print("Device set to: CPU")
 
 # Load CSV data
 df = pd.read_csv("iec_college_data.csv").dropna(subset=["content"]).reset_index(drop=True)
@@ -45,8 +45,8 @@ if collection.count() == 0:
             print(f"Indexed {idx}/{len(df)}")
     print("Indexing complete.")
 
-# QA model: Use a stronger model like Flan-T5-large if needed
-qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-base", device=device)
+# QA model: Use lighter model on CPU
+qa_pipeline = pipeline("text2text-generation", model="google/flan-t5-small", device=device)
 
 # QA function
 def answer_question(user_question):