Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load model and tokenizer
-model_name = "meta-llama/Meta-Llama-3-8B"
+model_name = "gpt2"  # or "gpt2-medium" for a larger model
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
@@ -16,8 +16,8 @@ iface = gr.Interface(
     fn=generate_text,
     inputs="text",
     outputs="text",
-    title="Meta-Llama-3-8B Text Generation",
-    description="Enter a prompt to generate text using Meta-Llama-3-8B."
+    title="GPT-2 Text Generation",
+    description="Enter a prompt to generate text using GPT-2."
 )
 
 iface.launch()