Update app.py

app.py

@@ -25,7 +25,7 @@ input_text = st.text_input("Prompt", "Write me a poem about Machine Learning.")
 # Generate text when the button is pressed
 if st.button("Generate"):
     input_ids = tokenizer(input_text, return_tensors="pt")
-    outputs = model.generate(**input_ids)
+    outputs = model.generate(**input_ids, max_new_tokens=1000)
     generated_text = tokenizer.decode(outputs[0])
     
     # Display the generated text