Update app.py

app.py

@@ -21,28 +21,21 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 
 class BasicAgent:
-    def __init__(self, model: str = "google/gemma-7b", searx_url: str = "https://search.example.org"):
-        self.client = InferenceClient(model=model)
-        self.searx_url = searx_url
-        print(f"Initialized with model: {model}")
-
+    def __init__(self, model: str = "google/gemma-7b"):
+        self.generator = pipeline(
+            "text-generation",
+            model=model,
+            device_map="auto"  # Uses GPU if available
+        )
+    
     def __call__(self, question: str) -> str:
-        print(f"Processing: {question[:50]}...")
-        return self.client.text_generation(
-            prompt=question,
+        outputs = self.generator(
+            question,
             max_new_tokens=100,
+            do_sample=True,
             temperature=0.7
         )
-
-    def web_search(self, query: str) -> List[Dict]:
-        """Search using SearxNG"""
-        params = {"q": query, "format": "json", "engines": "google,bing"}
-        try:
-            response = requests.get(self.searx_url, params=params, timeout=10)
-            return response.json().get("results", [])
-        except Exception as e:
-            print(f"Search failed: {e}")
-            return []
+        return outputs[0]['generated_text']
             
     def wikipedia_search(self, query: str) -> str:
         """Get Wikipedia summary"""