Update inference.py

inference.py

@@ -234,19 +234,19 @@ class Predictor:
         #output_file.close()
 
 
-        def transform_video(self, batch_size=4, start=0, end=0):
+    def transform_video(self, input_path, output_path, batch_size=4, start=0, end=0):
         end = end or None
 
-        # if not os.path.isfile(input_path):
-        #     raise FileNotFoundError(f'{input_path} does not exist')
+        if not os.path.isfile(input_path):
+            raise FileNotFoundError(f'{input_path} does not exist')
 
-        # output_dir = "/".join(output_path.split("/")[:-1])
-        # os.makedirs(output_dir, exist_ok=True)
-        # is_gg_drive = '/drive/' in output_path
-        # temp_file = ''
-        #
-        # if is_gg_drive:
-        #     temp_file = f'tmp_anime.{output_path.split(".")[-1]}'
+        output_dir = "/".join(output_path.split("/")[:-1])
+        os.makedirs(output_dir, exist_ok=True)
+        is_gg_drive = '/drive/' in output_path
+        temp_file = ''
+
+        if is_gg_drive:
+            temp_file = f'tmp_anime.{output_path.split(".")[-1]}'
 
         def transform_and_write(frames, count, writer):
             anime_images = self.transform(frames)
@@ -294,7 +294,7 @@ class Predictor:
         if temp_file:
             shutil.move(temp_file, output_path)
 
-        print(f'Animation video saved to {output_path}')    
+        print(f'Animation video saved to {output_path}')
 
     def transform_video1(self, video, batch_size, start, end):
         #end = end or None