Update app.py

app.py

@@ -14,23 +14,18 @@ from transformers import pipeline
 
 class BasicAgent:
     def __init__(self):
-        print("✅ AI Agent with transformers + PyTorch initialized")
-        self.generator = pipeline(
-            "text-generation", 
-            model="sshleifer/tiny-gpt2",  # or use "distilgpt2"
-            framework="pt"  # Explicitly use PyTorch
-        )
+        self.api_url = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.1"
+        self.headers = {
+            "Authorization": f"Bearer {os.getenv('HF_TOKEN')}"
+        }
 
     def __call__(self, question: str) -> str:
-        print(f"⚙️ Generating answer for: {question[:60]}...")
         try:
-            response = self.generator(question, max_length=50, num_return_sequences=1)
-            answer = response[0]['generated_text'].strip()
-            print(f"🧠 Answer: {answer}")
-            return answer
-        except Exception as e:
-            print(f"❌ Error in model inference: {e}")
-            return "Model error"
+            response = requests.post(self.api_url, headers=self.headers, json={"inputs": question})
+            return response.json()[0]["generated_text"].strip()
+        except:
+            return "API error"
+
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):