Update app.py

app.py

@@ -4,7 +4,7 @@ import uuid
 import subprocess
 from pathlib import Path
 from typing import Tuple
-
+from inference_img import run_imginference
 import gradio as gr
 
 # Demo images
@@ -14,7 +14,7 @@ DEMO = Path("demo/demo.gif")
 
 TMP_PREFIX = "/tmp/gradio/"
 TARGET_DIR = f"{TMP_PREFIX}output/"
-EXTENSION = "gif"
+EXTENSION = "png"
 
 def interpolate_image(img_a_path: str, img_b_path: str) -> Tuple[str, str]:
     """
@@ -32,12 +32,15 @@ def interpolate_image(img_a_path: str, img_b_path: str) -> Tuple[str, str]:
     palette_path = f"{TMP_PREFIX}palette.png"
 
     try:
+
+        run_imginference(img_a_path, img_b_path, exp=4)
+        
         # 1) Run interpolation (frames -> TARGET_DIR/img1.png ... imgN.png)
-        subprocess.run([
-            "python3", "inference_img.py",
-            "--img", str(img_a_path), str(img_b_path),
-            "--exp", "4"
-        ], check=True)
+        # # subprocess.run([
+        #   #   "python3", "inference_img.py",
+        #   #   "--img", str(img_a_path), str(img_b_path),
+        #       "--exp", "4"
+        # ], check=True)
 
         # 2) Generate palette
         subprocess.run([
@@ -48,12 +51,12 @@ def interpolate_image(img_a_path: str, img_b_path: str) -> Tuple[str, str]:
         ], check=True)
 
         # 3) Use palette to produce final GIF
+        #!ffmpeg -r 14 -f image2 -i output/img%d.png -vf "palettegen=stats_mode=single" palette.png
         subprocess.run([
-            "ffmpeg", "-y", "-r", "14", "-f", "image2",
+            "ffmpeg", "-y", "-r", "14", "-f",
             "-i", f"{TARGET_DIR}img%d.png",
-            "-i", palette_path,
-            "-lavfi", "paletteuse",
-            interpolated_path
+            str(palettegen=stats_mode=single)",
+            str(palette.png)
         ], check=True)
 
         return interpolated_image, interpolated_path