Merge branch 'main' of https://huggingface.co/spaces/vfontech/Multi-Input-Res-Diffusion-VFI

app.py

@@ -26,7 +26,7 @@ def to_numpy(img_tensor):
     img_np = np.clip(img_np, 0, 1)
     return (img_np * 255).astype(np.uint8)
 
-def interpolate(img0_pil, img2_pil, tau, num_samples):
+def interpolate(img0_pil, img2_pil, tau=0.5, num_samples=1):
     img0 = transform(img0_pil.convert("RGB")).unsqueeze(0).to(device)
     img2 = transform(img2_pil.convert("RGB")).unsqueeze(0).to(device)
 
@@ -70,10 +70,9 @@ demo = gr.Interface(
         "- If `Number of Samples` > 1, ignores Tau and generates a sequence of interpolated images."
     ),
     examples=[
-        ["_data/example_images/frame1.png", "_data/example_images/frame3.png", 0.5],
-    ],
+        ["_data/example_images/frame1.png", "_data/example_images/frame3.png", 0.5, 1],
+    ]
 )
 
 if __name__ == "__main__":
-    demo.queue(max_size=12)
-    demo.launch(max_threads=1)
+    demo.launch()

requirements.txt

@@ -1,7 +1,6 @@
 # Main dependencies
 torch>=2.6.0
 torchvision>=0.21.0
-lightning>=2.2.4
 numpy>=1.26.4
 matplotlib>=3.8.0
 pyyaml>=6.0.0
@@ -26,7 +25,6 @@ cupy-cuda12x>=12.0.0  # For CUDA 12.4
 
 # Utilities and tools
 scipy>=1.7.0
-tensorboard>=2.8.0
 
 # Project-Specific Dependencies
 # RAFT (Flow Estimation)