Update app.py

app.py

@@ -25,7 +25,7 @@ def classify_image(img):
     img = torch.unsqueeze(img, dim=0)
  
     # read class_indict
-    json_path = 'F:\mushroom_project\VIT\class_indices.json'
+    json_path = './class_indices.json'
     assert os.path.exists(json_path), "file: '{}' dose not exist.".format(json_path)
  
     with open(json_path, "r") as f:
@@ -34,9 +34,8 @@ def classify_image(img):
     # create model
     model = create_model(num_classes=370, has_logits=False).to(device)
     # load model weights
-    model_weight_path = "F:\mushroom_project\VIT\pretrain_30_weights\\best_model.pth"
-    #load no pretrain model path
-    #model_weight_path = "F:\mushroom_project\VIT\no_pretrain_weights\best_model.pth"
+    model_weight_path = "./best_model.pth"
+
     model.load_state_dict(torch.load(model_weight_path, map_location=device))
     model.eval()
     with torch.no_grad():