Update app.py

app.py

@@ -2,14 +2,12 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, Trainer, TrainingArguments
 from datasets import load_dataset
-import torch
-import os
 
 # -----------------------------
 # 1️⃣ Model setup
 # -----------------------------
 MODEL_DIR = "model"
-MODEL_NAME = "sshleifer/tiny-gpt2"  # tiny GPT-2, CPU-friendly
+MODEL_NAME = "distilgpt2"  # official public small GPT-2
 
 # Load tokenizer & model
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
@@ -79,4 +77,4 @@ with gr.Blocks() as demo:
         generate_output = gr.Textbox(label="Output")
         generate_button.click(generate_text, inputs=prompt_input, outputs=generate_output)
 
-demo.launch(share=True)
+demo.launch()