Update app.py

app.py

@@ -5,7 +5,7 @@
 import streamlit as st
 from PIL import Image
 import time
-from transformers import pipeline,AutoModelForCausalLM,AutoTokenizer
+from transformers import pipeline
 from typing import Tuple
 from datasets import load_dataset
 import soundfile as sf
@@ -83,7 +83,7 @@ def generate_story_content(system_prompt: str, user_prompt: str) -> str:
         # Prepare chat message structure
         conversation_history = [
             {"role": "system", "content": system_prompt},
-            {"role": "user", "content": user_prompt}
+            {"role": "user", "content": user_prompt},
         ]
 
         # Generate the story