Update app.py

app.py

@@ -15,7 +15,7 @@ from transformers import pipeline
 class BasicAgent:
     def __init__(self):
         print("Initializing BasicAgent with a transformer model...")
-        self.qa_pipeline = pipeline("text-generation", model="distilbert-base-uncased")  # You can change model if needed
+        self.qa_pipeline = pipeline("text-generation", model="sshleifer/tiny-distilroberta-base")  # You can change model if needed
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question: {question}")