Update app.py

app.py

@@ -20,7 +20,17 @@ class LocalLLM:
         self.pipe = pipe
 
     def generate(self, prompt, **kwargs):
-        return self.pipe(prompt)[0]["generated_text"]
+        try:
+            outputs = self.pipe(prompt, **kwargs)
+            # outputs = [{'generated_text': '...'}]
+            if isinstance(outputs, list) and isinstance(outputs[0], dict):
+                return outputs[0]["generated_text"]
+            elif isinstance(outputs, list):
+                return outputs[0]  # fallback if it's just a list of strings
+            return str(outputs)
+        except Exception as e:
+            print(f"❌ Error in LocalLLM.generate(): {e}")
+            raise
 
 @spaces.GPU
 def load_llm(hf_token):