Update app.py

app.py

@@ -2,8 +2,8 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load the model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("sagorsarker/emailgenerator")
-model = AutoModelForCausalLM.from_pretrained("sagorsarker/emailgenerator")
+tokenizer = AutoTokenizer.from_pretrained("postbot/gpt2-medium-emailgen")
+model = AutoModelForCausalLM.from_pretrained("postbot/gpt2-medium-emailgen")
 
 # Streamlit UI styling
 st.set_page_config(page_title="Email Generator", layout="centered")
@@ -35,13 +35,7 @@ st.markdown("""
         .button:hover {
             background-color: #4B0082;
         }
-        .generated-email {
-            background-color: #f4f4f4;
-            padding: 15px;
-            border-radius: 10px;
-            border: 1px solid #ddd;
-            font-family: 'Courier New', Courier, monospace;
-        }
+
     </style>
 """, unsafe_allow_html=True)
 
@@ -68,6 +62,6 @@ if generate_button:
         
         # Display the generated email with some styling
         st.markdown('<div class="header">Generated Email:</div>', unsafe_allow_html=True)
-        st.markdown(f'<div class="generated-email">{generated_email}</div>', unsafe_allow_html=True)
+        st.markdown(f'<div>{generated_email}</div>', unsafe_allow_html=True)
     else:
         st.error("Please enter a prompt to generate the email.")