Update app.py

app.py

@@ -1,14 +1,21 @@
 from text_generator import TextGenerationTool
 
-# Create an instance of the tool
-text_gen_tool = TextGenerationTool(default_model="gpt2")
+# Create an instance of the tool with a safer default model
+text_gen_tool = TextGenerationTool(default_model="distilgpt2")
 
 # Launch the Gradio interface
 if __name__ == "__main__":
     import gradio as gr
     
     with gr.Blocks(title="Text Generation Tool") as demo:
-        gr.Markdown("# Text Generation Tool")
+        # Add a warning about authentication
+        gr.Markdown("""
+        # Text Generation Tool
+        
+        > **Note:** This application can run without a Hugging Face token, but some models may require authentication.
+        > For best results with larger models, set the `HF_TOKEN` environment variable with your token.
+        """)
+
         
         with gr.Row():
             with gr.Column():
@@ -48,9 +55,9 @@ if __name__ == "__main__":
         
         gr.Examples(
             examples=[
-                ["Write a short story about a robot learning to paint.", "gpt2"],
-                ["Explain quantum computing to a 10-year-old.", "gpt2"],
-                ["Write a poem about the changing seasons.", "gpt2"]
+                ["Write a short story about a robot learning to paint.", "distilgpt2"],
+                ["Explain quantum computing to a 10-year-old.", "gpt2-small"],
+                ["Write a poem about the changing seasons.", "distilgpt2"]
             ],
             inputs=[prompt_input, model_dropdown]
         )