Update app.py

app.py

@@ -7,10 +7,10 @@ import os
 import spaces
 
 # Text Expansion Model (ZeroGPU compatible)
-text_generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
+text_generator = pipeline("text2text-generation", model="google/flan-t5-large")
 
-# Text-to-Speech Model (Bark small)
-tts_pipeline = pipeline("text-to-speech", model="suno/bark-small")
+# Text-to-Speech Model (ZeroGPU compatible)
+tts_pipeline = pipeline("text-to-speech", model="facebook/fastspeech2-en-ljspeech")
 
 # Image Generation Model (LoRA-based)
 image_generator = pipeline("text-to-image", model="stabilityai/sdxl-turbo")
@@ -20,19 +20,15 @@ image_generator = pipeline("text-to-image", model="stabilityai/sdxl-turbo")
 def generate_cartoon(script_text):
     # Step 1: Expand Script
     prompt = f"""
-    You are a cartoon script writer. Convert the following story into a detailed cartoon scene plan.
-    For each scene, provide:
-    1. Scene description (setting + action)
+    Convert the following cartoon story into a detailed scene plan in JSON format. For each scene, provide:
+    1. Scene description
     2. Dialogue
     3. Characters involved
     4. Background description
 
     Story:
-    """
     {script_text}
     """
-    Return result in JSON format.
-    """
 
     response = text_generator(prompt, max_new_tokens=1024)[0]['generated_text']