Update app.py

app.py

@@ -19,8 +19,12 @@ def suggest_keywords(prompt):
         return "Model failed to load. Please check the logs or environment."
     
     try:
-        results = keyword_generator(prompt, max_length=50, num_return_sequences=3)
-        suggestions = [res["text"].strip() for res in results]
+        # Generate keywords
+        results = keyword_generator(prompt, max_length=50, num_return_sequences=3, truncation=True, pad_token_id=keyword_generator.tokenizer.eos_token_id)
+        
+        # Fix to extract 'generated_text'
+        suggestions = [res["generated_text"].strip() for res in results]
+        
         return "\n".join(suggestions)
     except Exception as e:
         return f"Error generating keywords: {e}"