Update app.py

app.py

@@ -20,8 +20,10 @@ class LocalLLM:
         self.pipe = pipe
 
     def generate(self, prompt, **kwargs):
+        unsupported_keys = ["stop_sequences"] # Remove keys not accepted by HF pipelines
+        cleaned_kwargs = {k: v for k, v in kwargs.items() if k not in unsupported_keys}
         try:
-            outputs = self.pipe(prompt, **kwargs)
+            outputs = self.pipe(prompt, **cleaned_kwargs)
             # outputs = [{'generated_text': '...'}]
             if isinstance(outputs, list) and isinstance(outputs[0], dict):
                 return outputs[0]["generated_text"]