Update predict.py

predict.py

@@ -57,4 +57,30 @@ def gradio_predict(audio, bg_color, fg_alpha, bars_color, bar_count, bar_width,
     predictor = Predictor()
     result = predictor.predict(
         audio=audio,
-        bg_color=bg 
+        bg_color=bg_color,
+        fg_alpha=fg_alpha,
+        bars_color=bars_color,
+        bar_count=bar_count,
+        bar_width=bar_width,
+        caption_text=caption_text
+    )
+    return result
+
+# Launch Gradio interface
+interface = gr.Interface(
+    fn=gradio_predict,
+    inputs=[
+        gr.Audio(source="upload", type="filepath", label="Audio File"),
+        gr.Textbox(value="#000000", label="Background Color"),
+        gr.Slider(0, 1, value=0.75, label="Foreground Opacity"),
+        gr.ColorPicker(value="#ffffff", label="Bars Color"),
+        gr.Slider(10, 500, value=100, step=1, label="Number of Bars"),
+        gr.Slider(0, 1, value=0.4, step=0.1, label="Bar Width"),
+        gr.Textbox(default="", label="Caption Text")
+    ],
+    outputs=gr.Video(label="Waveform Video"),
+    live=False
+)
+
+if __name__ == "__main__":
+    interface.launch(server_name="0.0.0.0", server_port=7860)