Update predict.py

predict.py

@@ -1,21 +1,21 @@
 import subprocess
-from cog import BasePredictor, Input, Path
 import gradio as gr
 import tempfile
+from pathlib import Path
 
-class Predictor(BasePredictor):
+class Predictor:
     def predict(self,
-        audio: Path = Input(description="Audio file to create waveform from"),
-        bg_color: str = Input(description="Background color of waveform", default="#000000"),
-        fg_alpha: float = Input(description="Opacity of foreground waveform", default=0.75),
-        bars_color: str = Input(description="Color of waveform bars", default="#ffffff"),
-        bar_count: int = Input(description="Number of bars in waveform", default=100),
-        bar_width: float = Input(description="Width of bars in waveform. 1 represents full width, 0.5 represents half width, etc.", default=0.4),
-        caption_text: str = Input(description="Caption text for the video", default=""),
-    ) -> Path:
+        audio: str,
+        bg_color: str = "#000000",
+        fg_alpha: float = 0.75,
+        bars_color: str = "#ffffff",
+        bar_count: int = 100,
+        bar_width: float = 0.4,
+        caption_text: str = "",
+    ) -> str:
         """Make waveform video from audio file"""
         waveform_video = gr.make_waveform(
-            str(audio),
+            audio,
             bg_color=bg_color,
             fg_alpha=fg_alpha,
             bars_color=bars_color,
@@ -24,7 +24,7 @@ class Predictor(BasePredictor):
         )
 
         if caption_text == "" or caption_text is None:
-            return Path(waveform_video)
+            return waveform_video
         else:
             padded_waveform_path = tempfile.mktemp(suffix=".mp4")
             background_image_path = tempfile.mktemp(suffix=".png")
@@ -50,7 +50,7 @@ class Predictor(BasePredictor):
                 '-filter_complex', 'overlay=0:0', final_video_path
             ], check=True)
 
-        return Path(final_video_path)
+        return final_video_path
 
 # Gradio user interface
 def gradio_predict(audio, bg_color, fg_alpha, bars_color, bar_count, bar_width, caption_text):