Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 import spaces
 
-import torch
-torch.jit.script = lambda f: f
+
 # torch.autocast = lambda device_type, dtype: torch.autocast(device_type, torch.float)
 
 # Initialize the model only once
@@ -12,6 +11,8 @@ torch.jit.script = lambda f: f
 
 @spaces.GPU
 def generate(model_name, image, text):
+    import torch
+    torch.jit.script = lambda f: f
     
     from t2v_metrics import VQAScore, list_all_vqascore_models