Update app.py

app.py

@@ -164,7 +164,7 @@ def main():
     global step
     global dtype
 
-    device = "auto"
+    device = "cuda"
     dtype = torch.float16
     result=[]
     step = 2