added files

app.py

@@ -9,6 +9,16 @@ from rembg import remove
 from PIL import Image, ImageOps
 from predict_pose import generate_pose_keypoints
 
+
+# Use GPU if available
+if torch.cuda.is_available():
+    device = torch.device("cuda")
+else:
+    device = torch.device("cpu")
+
+print(f"####Using {device}.#####") 
+
+
 # Make directories
 os.system("mkdir ./Data_preprocessing")
 os.system("mkdir ./Data_preprocessing/test_color")
@@ -126,4 +136,4 @@ gr.Interface(
     allow_flagging=False,
     analytics_enabled=False,
     thumbnail=thumbnail,
-    ).launch(debug=True, enable_queue=True)
+    ).launch(debug=True, enable_queue=True)