torch.jit.script = lambda f: f

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import spaces
 import numpy as np
 import torch
+torch.jit.script = lambda f: f
 import cv2
 import os
 import imageio