Hugging Face's logo

Spaces:
Shakir60
/
Test
Sleeping

App Files Community
Shakir60 commited on
Commit
3289f09
·
verified ·
1 Parent(s): 4215f31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ st.set_page_config(
24
  def load_models():
25
  # Vision model
26
  from huggingface_hub import login
27
- login(token="HUGGINGFACE_TOKEN")
28
  damage_model = ViTForImageClassification.from_pretrained(
29
  "microsoft/vit-base-patch16-224",
30
  use_auth_token=True
 
24
  def load_models():
25
  # Vision model
26
  from huggingface_hub import login
27
+ login(token="HUGGINGFACE_API_TOKEN")
28
  damage_model = ViTForImageClassification.from_pretrained(
29
  "microsoft/vit-base-patch16-224",
30
  use_auth_token=True