Update app.py

app.py

@@ -28,7 +28,7 @@ def generate_poem(theme, style, keywords):
 
     prompt = f"Write a poem about {theme} in the style of {style} using these keywords: {keywords}"
     #output = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
-    output = llm.invoke(prompt)
+    output = llm.invoke(prompt, max_length=100, num_return_sequences=1)
     return output
 
 submit = st.button("Enter to generate the Poem")