Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import torch
 import tempfile
 from huggingface_hub import snapshot_download
 import sys, os