Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 import os
 
 # Load the pre-trained language model and tokenizer from Hugging Face
-model_name = "gpt2-large"  # Use "EleutherAI/gpt-j-6B" for GPT-J (requires more resources)
+model_name = "gpt2-medium"  # Use "EleutherAI/gpt-j-6B" for GPT-J (requires more resources)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
@@ -18,7 +18,7 @@ def generate_response(prompt):
     trump_prompt = f"Donald Trump says: {prompt}"
     
     # Generate the response
-    response = generator(trump_prompt, max_length=100, num_return_sequences=1, temperature=0.8)
+    response = generator(trump_prompt, max_length=50, num_return_sequences=1, temperature=0.8)
     return response[0]['generated_text']
 
 # Function to convert text to audio using gTTS