Update predict.py
Browse files- predict.py +27 -1
predict.py
CHANGED
@@ -57,4 +57,30 @@ def gradio_predict(audio, bg_color, fg_alpha, bars_color, bar_count, bar_width,
|
|
57 |
predictor = Predictor()
|
58 |
result = predictor.predict(
|
59 |
audio=audio,
|
60 |
-
bg_color=
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
predictor = Predictor()
|
58 |
result = predictor.predict(
|
59 |
audio=audio,
|
60 |
+
bg_color=bg_color,
|
61 |
+
fg_alpha=fg_alpha,
|
62 |
+
bars_color=bars_color,
|
63 |
+
bar_count=bar_count,
|
64 |
+
bar_width=bar_width,
|
65 |
+
caption_text=caption_text
|
66 |
+
)
|
67 |
+
return result
|
68 |
+
|
69 |
+
# Launch Gradio interface
|
70 |
+
interface = gr.Interface(
|
71 |
+
fn=gradio_predict,
|
72 |
+
inputs=[
|
73 |
+
gr.Audio(source="upload", type="filepath", label="Audio File"),
|
74 |
+
gr.Textbox(value="#000000", label="Background Color"),
|
75 |
+
gr.Slider(0, 1, value=0.75, label="Foreground Opacity"),
|
76 |
+
gr.ColorPicker(value="#ffffff", label="Bars Color"),
|
77 |
+
gr.Slider(10, 500, value=100, step=1, label="Number of Bars"),
|
78 |
+
gr.Slider(0, 1, value=0.4, step=0.1, label="Bar Width"),
|
79 |
+
gr.Textbox(default="", label="Caption Text")
|
80 |
+
],
|
81 |
+
outputs=gr.Video(label="Waveform Video"),
|
82 |
+
live=False
|
83 |
+
)
|
84 |
+
|
85 |
+
if __name__ == "__main__":
|
86 |
+
interface.launch(server_name="0.0.0.0", server_port=7860)
|