Update app.py

app.py

@@ -111,7 +111,7 @@ def main():
 
         progress_bar_image.progress(7)
 
-        device = "gpu" if torch.cuda.is_available() else "cpu"
+        device = "cuda" if torch.cuda.is_available() else "cpu"
 
         print("used device is", device)
         pipe.to(device)