Update app.py

app.py

@@ -18,7 +18,7 @@ import folium
 token = os.environ.get("token")
 
 local_dir = snapshot_download(
-    repo_id="robocan/GeoG-GCP",
+    repo_id="robocan/GeoG_23k",
     repo_type="model",
     local_dir="SVD",
     token=token
@@ -35,9 +35,11 @@ class ModelPre(torch.nn.Module):
         self.embedding = torch.nn.Sequential(
             *list(models.convnext_small(weights=models.ConvNeXt_Small_Weights.IMAGENET1K_V1).children())[:-1],
             torch.nn.Flatten(),
-            torch.nn.Linear(in_features=768, out_features=512),
+            torch.nn.Linear(in_features=768, out_features=1024),
             torch.nn.ReLU(),
-            torch.nn.Linear(in_features=512, out_features=len_classes),
+            torch.nn.Linear(in_features=1024, out_features=1024),
+            torch.nn.ReLU(),
+            torch.nn.Linear(in_features=1024, out_features=len_classes),
         )
 
     def forward(self, data):