Update app.py

app.py

@@ -589,7 +589,7 @@ import gradio as gr
 from transformers import pipeline
 
 # Load a lightweight text generation model from Hugging Face
-generator = pipeline("text-generation", model="distilgpt2", max_length=100)
+generator = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", max_length=100)
 
 # Initialize story state
 story_history = []