Upload predict.py

predict.py

@@ -49,7 +49,7 @@ class Predictor(BasePredictor):
         index: int = Input(description="video Source", default=0),
         target: CogPath = Input(description="face image", default=None),
         output_quality: int = Input(description="WARNING!!! 0 - HIGHEST 100 - LOWEST", default=0),
-    ) -> Path:
+    ) -> CogPath:
         out_path = Path(tempfile.mkdtemp(suffix='.mp4'))
         print('*****************************************>')
         print(out_path)
@@ -66,4 +66,4 @@ class Predictor(BasePredictor):
         out_path = output_dir + video_name
         run_cmd("python run.py --target "+source+" --source "+target+" -o "+str(out_path)+f" --execution-provider cuda --frame-processor face_swapper --output-video-quality {int(output_quality)}")
         # output_file = output_dir + "/"+video_name
-        return Path(out_path)
+        return CogPath(out_path)