Update app.py

app.py

@@ -18,7 +18,7 @@ class BasicAgent:
         # initialize HF inference pipeline once
         if HF_TOKEN is None:
             raise ValueError("HF_TOKEN not set in environment")
-        self.generator = pipeline("text-generation", model="distilbert/distilgpt2")
+        self.generator = pipeline("text-generation", model="bigscience/bloom-560m")
         # The GAIA system prompt (no "FINAL ANSWER:" at the end)
         self.system_prompt = (
             "You are a general AI assistant. "
@@ -34,7 +34,7 @@ class BasicAgent:
         # Run the model
         out = self.generator(
             prompt,
-            max_new_tokens=16,       # leave room for the answer
+            max_new_tokens=50,       # leave room for the answer
             return_full_text=False
         )
         answer = out[0]["generated_text"].strip()