Update app.py

app.py

@@ -6,11 +6,10 @@ from moviepy.editor import ImageSequenceClip, VideoFileClip, AudioFileClip
 import numpy as np
 import os
 from mutagen.mp3 import MP3
-import cv2
+import soundfile as sf
 from dotenv import load_dotenv
-from transformers import pipeline, AutoProcessor, AutoModel
+from transformers import AutoProcessor, AutoModel
 import torch
-import soundfile as sf
 
 # Load environment variables
 load_dotenv()
@@ -57,33 +56,36 @@ def merge_audio_video(entities_num, resize_img_list, text_input):
     
     return mergedclip
 
-def engine(text_input):
+with gr.Blocks() as app:
+    # Load models in Blocks context
     ner = gr.Interface.load("huggingface/flair/ner-english-ontonotes-large")
-    entities = ner(text_input)
-    entities = [tupl for tupl in entities if None not in tupl]
-    entities_num = len(entities)
-    img_list = []
-    
     latentdiffusion = gr.Interface.load("spaces/multimodalart/latentdiffusion")
-    
-    for ent in entities:
-        img = latentdiffusion(ent[0], '50', '256', '256', '1', 10)[0]
-        img_list.append(img)
-    
-    resize_img_list = resize(img_list)
-    mergedclip = merge_audio_video(entities_num, resize_img_list, text_input)
-    mergedclip.write_videofile('mergedvideo.mp4')
-    
-    return 'mergedvideo.mp4'
 
-app = gr.Interface(
-    fn=engine, 
-    inputs=gr.Textbox(lines=5, label="Input Text"), 
-    outputs=gr.Video(label='Final Merged Video'), 
-    description="<div>🎭🎞️🍿 AI Movie Maker - Comedy 🎬 🧠 🎨</div>",
-    examples=[
-        ["Two space marines take up arms to save the planet from an alien invasion. These two dashing strong men play a comedic role in the science fiction movie of the future where even Barnaby bunny is willing to join their wacky gang of space marines to save the planet with good looks and comedy."]
-    ],
-    title="AI Pipeline Multi Model 🎭🎞️🍿 Movie Maker 🎬 🧠 🎨",
-    article="<br><div></div>"
-).launch(debug=True)
+    def engine(text_input):
+        entities = ner(text_input)
+        entities = [tupl for tupl in entities if None not in tupl]
+        entities_num = len(entities)
+        img_list = []
+        
+        for ent in entities:
+            img = latentdiffusion(ent[0], '50', '256', '256', '1', 10)[0]
+            img_list.append(img)
+        
+        resize_img_list = resize(img_list)
+        mergedclip = merge_audio_video(entities_num, resize_img_list, text_input)
+        mergedclip.write_videofile('mergedvideo.mp4')
+        
+        return 'mergedvideo.mp4'
+
+    interface = gr.Interface(
+        fn=engine,
+        inputs=gr.Textbox(lines=5, label="Input Text"),
+        outputs=gr.Video(label='Final Merged Video'),
+        description="<div>🎭🎞️🍿 AI Movie Maker - Comedy 🎬 🧠 🎨</div>",
+        examples=[
+            ["Two space marines take up arms to save the planet from an alien invasion. These two dashing strong men play a comedic role in the science fiction movie of the future where even Barnaby bunny is willing to join their wacky gang of space marines to save the planet with good looks and comedy."]
+        ],
+        title="AI Pipeline Multi Model 🎭🎞️🍿 Movie Maker 🎬 🧠 🎨",
+        article="<br><div></div>"
+    )
+    interface.launch(debug=True)