Update app.py

app.py

@@ -18,14 +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="gpt2",
-            tokenizer="gpt2",
-            framework="pt",             # force PyTorch
-            token=HF_TOKEN,
-            device=-1
-        )
+        self.generator = pipeline("text-generation", model="distilbert/distilgpt2")
         # The GAIA system prompt (no "FINAL ANSWER:" at the end)
         self.system_prompt = (
             "You are a general AI assistant. "