app.py: try-catch cuda bug when try to execte the model

app.py

@@ -21,7 +21,7 @@ transform = Compose([
     Normalize(mean=[0.5]*3, std=[0.5]*3),
 ])
 
-def to_numpy(img_tensor):
+def to_numpy(img_tensor: torch.Tensor) -> np.ndarray:
     img_np = denorm(img_tensor, mean=[0.5]*3, std=[0.5]*3).squeeze().permute(1, 2, 0).cpu().numpy()
     img_np = np.clip(img_np, 0, 1)
     return (img_np * 255).astype(np.uint8)
@@ -30,21 +30,25 @@ 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)
 
-    if num_samples == 1:
-        # Unique image
-        img1 = model.reverse_process([img0, img2], tau)
-        return Image.fromarray(to_numpy(img1)), None
-    else:
-        # Múltiples imágenes → video
-        frames = [to_numpy(img0)]
-        for t in np.linspace(0, 1, num_samples):
-            img = model.reverse_process([img0, img2], float(t))
-            frames.append(to_numpy(img))
-        frames.append(to_numpy(img2))
+    try:
+        if num_samples == 1:
+            # Unique image
+            img1 = model.reverse_process([img0, img2], tau)
+            return Image.fromarray(to_numpy(img1)), None
+        else:
+            # Múltiples imágenes → video
+            frames = [to_numpy(img0)]
+            for t in np.linspace(0, 1, num_samples):
+                img = model.reverse_process([img0, img2], float(t))
+                frames.append(to_numpy(img))
+            frames.append(to_numpy(img2))
 
-        temp_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
-        imageio.mimsave(temp_path, frames, fps=8)
-        return None, temp_path
+            temp_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
+            imageio.mimsave(temp_path, frames, fps=8)
+            return None, temp_path
+    except Exception as e:
+        print(f"Error during interpolation: {e}")
+        return None, None
 
 demo = gr.Interface(
     fn=interpolate,