Update app.py

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the conversational model
+# Load a text generation model
 @st.cache_resource
 def load_model():
-    return pipeline("text-generation", model="facebook/blenderbot-400M-distill")
+    return pipeline("text-generation", model="EleutherAI/gpt-neo-125M")
 
 # Initialize the model
 generator = load_model()
@@ -12,16 +12,23 @@ generator = load_model()
 st.title("Donald Trump Style Text Generator")
 st.write("Enter a prompt, and the model will respond in a manner inspired by Donald Trump.")
 
-# Input from the user
+# User input
 user_input = st.text_input("Enter your prompt:")
 
 if user_input:
-    # Prefix to encourage Trump-inspired responses
+    # Add a prompt prefix to suggest the tone
     trump_prompt = f"In the style of Donald Trump: {user_input}"
     
-    # Generate the conversational response
-    conversation = generator(trump_prompt)
-    
-    # Extract and display the generated text
-    generated_text = conversation.generated_responses[0]
+    # Generate the response
+    response = generator(
+        trump_prompt, 
+        max_length=100, 
+        num_return_sequences=1, 
+        do_sample=True, 
+        temperature=0.8, 
+        top_p=0.9
+    )
+
+    # Extract the generated text from the response
+    generated_text = response[0]["generated_text"]
     st.write("Response:", generated_text)