wip

stayallive · May 20, 2023 · eec2075 · eec2075
1 parent 7d34900
commit eec2075
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/predict.py b/predict.py
@@ -66,9 +66,8 @@ def predict(
 
         return ModelOutput(
             text=result["text"],
-            language=LANGUAGES[result["language"]],
-            srt_path=out_path_srt,
-            vtt_path=out_path_vtt,
+            srt_file=out_path_srt,
+            vtt_file=out_path_vtt,
         )