Update app.py

app.py

@@ -4,10 +4,16 @@ import cv2
 import numpy as np
 import time
 import random
-
+import torch
 from PIL import Image
 from transparent_background import Remover
 
+@torch.jit.script
+def hard_sigmoid_jit_bwd(x: torch.Tensor, grad_output: torch.Tensor) -> torch.Tensor:
+    m = torch.ones_like(x) * ((x >= -3.) & (x <= 3.)) / 6.
+    return grad_output * m
+
+
 @spaces.GPU  # 修改装饰器，去掉括号
 def doo(video, mode):
     if(mode == 'Fast'):