Update app.py

app.py

@@ -15,7 +15,11 @@ if email_content:
     prompt = f"Generate a professional email reply to the following email: {email_content}"
 
     # Generate the reply using the model
-    generated_reply = pipe(prompt)
-    
+    generated_reply = pipe(prompt, max_length=200, num_return_sequences=1)
+
     # Display the generated reply
-    st.write(generated_reply[0]['generated_text'])
+    if generated_reply:
+        # Show only the generated text
+        st.write(generated_reply[0]['generated_text'].strip())
+    else:
+        st.write("Sorry, the model could not generate a reply.")