Update app.py

app.py

@@ -35,7 +35,7 @@ generator = pipeline(
 # Function to generate stories
 def generate_story(prompt, max_tokens=300, temperature=0.8):
     try:
-        chat_prompt = to_chat_prompt(prompt)
+        chat_prompt = chat_prompt(prompt)
         outputs = generator(
             chat_prompt,
             max_new_tokens=int(max_tokens),