Update app.py

app.py

@@ -6,11 +6,11 @@ import torch
 import os
 
 # Load the pre-trained language model and tokenizer from Hugging Face
-model_name = "gpt2-medium"  # Replace with a specific Trump-like model if available on Hugging Face
+model_name = "gpt2-large"  # Use "EleutherAI/gpt-j-6B" for GPT-J (requires more resources)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# Set up a text generation pipeline
+# Set up a text generation pipeline with the larger model
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)
 
 # Function to generate Trump-like response