Update app.py

app.py

@@ -30,12 +30,13 @@ def load_models():
     # Story generation model (Qwen3-1.7B)
     storyteller = pipeline(
         "text-generation",
-        model="Qwen/Qwen3-1.7B",
+        model="Qwen/Qwen3-0.5B",
         device_map="auto",
         trust_remote_code=True,
         torch_dtype="auto",
-        max_new_tokens=250,
-        temperature=0.7,
+        max_new_tokens=230,
+        temperature=0.8,
+        top_k=50,
         top_p=0.85,
         repetition_penalty=1.15,
         eos_token_id=151645