Update app.py

app.py

@@ -1,30 +1,27 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the Hugging Face model
-generator = pipeline("text-generation", model="gpt-2")
+# Load the conversational model
+@st.cache_resource
+def load_model():
+    return pipeline("conversational", model="facebook/blenderbot-400M-distill")
+
+# Initialize the model
+generator = load_model()
 
-# Streamlit app setup
 st.title("Donald Trump Style Text Generator")
 st.write("Enter a prompt, and the model will respond in a manner inspired by Donald Trump.")
 
-# User input
+# Input from the user
 user_input = st.text_input("Enter your prompt:")
 
 if user_input:
-    # Prefix the prompt for stylistic guidance
+    # Prefix to encourage Trump-inspired responses
     trump_prompt = f"In the style of Donald Trump: {user_input}"
     
-    # Generate response with conversational tuning
-    response = generator(
-        trump_prompt, 
-        max_length=100, 
-        num_return_sequences=1, 
-        do_sample=True, 
-        temperature=0.8, 
-        top_p=0.9
-    )
-
-    # Display the response
-    generated_text = response[0]["generated_text"]
+    # Generate the conversational response
+    conversation = generator(trump_prompt)
+    
+    # Extract and display the generated text
+    generated_text = conversation.generated_responses[0]
     st.write("Response:", generated_text)