Update app.py

app.py

@@ -5,6 +5,7 @@ from typing import Dict, Any, List, Tuple, Optional
 
 import gradio as gr
 import spaces
+import torch
 
 from torch.serialization import add_safe_globals
 from torch.torch_version import TorchVersion