Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -15,10 +15,8 @@ from huggingface_hub import snapshot_download
|
|
15 |
import s2sphere
|
16 |
import folium
|
17 |
|
18 |
-
# Retrieve the token from the environment variables
|
19 |
token = os.environ.get("token")
|
20 |
|
21 |
-
# Download the repository snapshot
|
22 |
local_dir = snapshot_download(
|
23 |
repo_id="robocan/GeoG-GCP",
|
24 |
repo_type="model",
|
@@ -41,7 +39,6 @@ class ModelPre(torch.nn.Module):
|
|
41 |
torch.nn.ReLU(),
|
42 |
torch.nn.Linear(in_features=512, out_features=len_classes),
|
43 |
)
|
44 |
-
# Freeze all layers
|
45 |
|
46 |
def forward(self, data):
|
47 |
return self.embedding(data)
|
|
|
15 |
import s2sphere
|
16 |
import folium
|
17 |
|
|
|
18 |
token = os.environ.get("token")
|
19 |
|
|
|
20 |
local_dir = snapshot_download(
|
21 |
repo_id="robocan/GeoG-GCP",
|
22 |
repo_type="model",
|
|
|
39 |
torch.nn.ReLU(),
|
40 |
torch.nn.Linear(in_features=512, out_features=len_classes),
|
41 |
)
|
|
|
42 |
|
43 |
def forward(self, data):
|
44 |
return self.embedding(data)
|