Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-# Load the pre-trained WormGPT model and tokenizer
-model = GPT2LMHeadModel.from_pretrained("wormgpt")
-tokenizer = GPT2Tokenizer.from_pretrained("wormgpt")
+# Load the pre-trained GPT2 model and tokenizer
+model = GPT2LMHeadModel.from_pretrained("gpt2")
+tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 
 def generate_text(prompt, max_length=50):
     input_ids = tokenizer.encode(prompt, return_tensors="pt")
@@ -19,8 +19,8 @@ iface = gr.Interface(
     fn=predict,
     inputs=gr.Textbox(lines=2, placeholder="Enter your text here..."),
     outputs="text",
-    title="WormGPT Text Generation",
-    description="Generate text using the WormGPT model."
+    title="GPT2 Text Generation",
+    description="Generate text using the GPT2 model."
 )
 
 iface.launch()