Update app.py

app.py

@@ -1,8 +1,13 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, GPT2Tokenizer, GPT2LMHeadModel
 
-# Load the text generation model
-generator = pipeline('text-generation', model='gpt2')
+# Pre-download the model and tokenizer
+model_name = "gpt2"
+GPT2Tokenizer.from_pretrained(model_name)
+GPT2LMHeadModel.from_pretrained(model_name)
+
+# Load the pipeline
+generator = pipeline('text-generation', model=model_name)
 
 # Function to generate text
 def generate_text(prompt):
@@ -18,5 +23,5 @@ iface = gr.Interface(
     description="Generate text with AI!"
 )
 
-# Launch it now
+# Launch it with model!
 iface.launch()