Add application file

core.py

@@ -13,11 +13,11 @@ def load_model(pretrained_dict, new):
     new.load_state_dict(model_dict)
 
 
-if torch.cuda.is_available():
-    device = torch.device("cuda")
-else:
-    device = torch.device("cpu")
-    print("use cpu")
+# if torch.cuda.is_available():
+#     device = torch.device("cuda")
+# else:
+device = torch.device("cpu")
+print("use cpu")
 
 # model_ckpt_path = "./models/resnet18.pth"
 model_ckpt_path = "https://huggingface.co/M4869/beauty_prediction_fpb5k/blob/main/resnet18.pth"