robocan commited on
Commit
5679c26
1 Parent(s): 0e4e27b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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)