Update app.py

app.py

@@ -50,7 +50,7 @@ def audio_video():
     
     return "results/final_output.mp4"
 
-#@spaces.GPU
+@spaces.GPU
 def run_inference(input_image, input_audio=None, input_text=None):
     pil_image = Image.fromarray(input_image.astype(np.uint8))
 
@@ -125,14 +125,14 @@ def run():
         }
     """
 
-    with gr.Blocks(theme=gr.themes.Default(), css=custom_css) as demo:
-        with gr.Column(elem_id="container"):
+    with gr.Blocks(css=custom_css) as demo:
+        with gr.Group():   
             gr.Markdown("""
                 # Talkie
                 <div class='subtitle'>Upload an image, add some audio or text, and watch the magic happen! ✨</div>
                 """)
-
-            input_image = gr.Image(label="📸 Your image")
+            with gr.Row():
+                input_image = gr.Image(label="📸 Your image")
 
             with gr.Row():
                 input_audio = gr.Audio(label="🎵 Your audio (Optional)")
@@ -140,16 +140,16 @@ def run():
                     label="💭 Your text",
                     placeholder="Type your text here..."
                 )
+            with gr.Row():
+                btn = gr.Button("GENERATE", elem_classes=["generate-btn"])
 
-            btn = gr.Button("GENERATE", elem_classes=["generate-btn"])
-
-            video_out = gr.Video(label="🎥 Your video", show_label=True)
-
-            btn.click(fn=run_inference, inputs=[input_image, input_audio, input_text], outputs=video_out)
+            with gr.Row():
+                video_out = gr.Video(label="🎥 Your video", show_label=True)
 
+        
+        btn.click(fn=run_inference, inputs=[input_image, input_audio, input_text], outputs=video_out)
         demo.queue()
-
-    return demo.launch()
+        demo.launch(server_name="0.0.0.0", server_port=7860)
 
 if __name__ == "__main__":
     run()