Update app.py

app.py

@@ -2,23 +2,22 @@ import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from gtts import gTTS
 import torch
-import os
 
-# Load the pre-trained language model and tokenizer from Hugging Face
-model_name = "gpt2-medium"  # Use "EleutherAI/gpt-j-6B" for GPT-J (requires more resources)
+# Load DialoGPT model for conversational style
+model_name = "microsoft/DialoGPT-medium"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# Set up a text generation pipeline with the larger model
+# Set up a text generation pipeline with the conversational model
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)
 
 # Function to generate Trump-like response
 def generate_response(prompt):
     # Craft the prompt to encourage a Trump-like response
-    trump_prompt = f"Donald Trump says: {prompt}"
+    trump_prompt = f"Imagine you are Donald Trump. Respond in your style to: {prompt}"
     
     # Generate the response
-    response = generator(trump_prompt, max_length=50, num_return_sequences=1, temperature=0.8)
+    response = generator(trump_prompt, max_length=50, num_return_sequences=1, temperature=0.6)
     return response[0]['generated_text']
 
 # Function to convert text to audio using gTTS