Update app.py

app.py

@@ -28,7 +28,7 @@ local_dir = snapshot_download(
 
 device = 'cpu'
 le = LabelEncoder()
-le = joblib.load("le.gz")
+le = joblib.load("SVD/le.gz")
 len_classes = len(le.classes_) + 1
 
 class ModelPre(torch.nn.Module):
@@ -49,7 +49,7 @@ class ModelPre(torch.nn.Module):
 # Load the pretrained model 
 model = ModelPre()
 
-model_w = torch.load("GeoG.pth", map_location=torch.device(device))
+model_w = torch.load("SVD/GeoG.pth", map_location=torch.device(device))
 model.load_state_dict(model_w['model'])
 
 cmp = transforms.Compose([