Update app.py

app.py

@@ -7,6 +7,7 @@ import cv2
 import huggingface_hub
 import subprocess
 import threading
+import torch
 from subprocess import getoutput
 
 is_shared_ui = False