Update app.py

app.py

@@ -18,6 +18,7 @@ from diffusers.utils import load_image
 
 predictor = None
 task_type = None
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 def init_predictor():
     global predictor