Update app.py

app.py

@@ -21,9 +21,9 @@ def run_async(coro):
     loop.close()
     return result
 
-# Load local text generation model
+# Load lightweight text generation model
 try:
-    generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2", device="cpu")
+    generator = pipeline("text-generation", model="distilbert/distilgpt2", device="cpu")
 except Exception as e:
     print(f"Error loading model: {e}")
     generator = None
@@ -46,7 +46,7 @@ def generate_script(prompt, custom_text=None):
     
     # Generate script with local model
     if generator:
-        input_text = f"Genera un guion para un video sobre '{prompt}'. Crea una lista numerada con descripciones breves (máximo 20 palabras por ítem) para un top 10 relacionado con el tema."
+        input_text = f"Generate a script for a video about '{prompt}'. Create a numbered list with short descriptions (max 20 words per item) for a top 10 related to the topic."
         try:
             result = generator(input_text, max_length=300, num_return_sequences=1, do_sample=True)[0]['generated_text']
             return result.strip()
@@ -129,7 +129,7 @@ def create_video(prompt, custom_text, music_file):
     else:
         final_clip = final_clip.set_audio(audio)
 
-    # Write final video (no subtitles to avoid ImageMagick error)
+    # Write final video
     final_clip.write_videofile(output_video, codec="libx264", audio_codec="aac", fps=24)
     
     # Clean up