adwod commited on
Commit
3db0bd7
1 Parent(s): f76e3fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,12 +4,12 @@ from transformers import ViTFeatureExtractor, ViTForImageClassification, ViTConf
4
  from huggingface_hub import hf_hub_url, cached_download
5
 
6
  # URL del archivo de configuraci贸n en el espacio de Hugging Face
7
- config_url = hf_hub_url(filename="config.json", repo_id=""adwod/Streamlite_ViT_2000"")
8
  # Descargar el archivo de configuraci贸n y cargarlo en una instancia de ViTConfig
9
  config = ViTConfig.from_json_dict(cached_download(config_url))
10
 
11
  # URL del archivo de pesos del modelo en el espacio de Hugging Face
12
- model_path = hf_hub_url(filename="pytorch_model.bin", repo_id=""adwod/Streamlite_ViT_2000"")
13
  # Descargar el archivo de pesos del modelo y cargarlo en el modelo
14
  model.load_state_dict(torch.load(cached_download(model_path)))
15
  model = ViTForImageClassification(config)
 
4
  from huggingface_hub import hf_hub_url, cached_download
5
 
6
  # URL del archivo de configuraci贸n en el espacio de Hugging Face
7
+ config_url = hf_hub_url(filename="config.json", repo_id="adwod/Streamlite_ViT_2000")
8
  # Descargar el archivo de configuraci贸n y cargarlo en una instancia de ViTConfig
9
  config = ViTConfig.from_json_dict(cached_download(config_url))
10
 
11
  # URL del archivo de pesos del modelo en el espacio de Hugging Face
12
+ model_path = hf_hub_url(filename="pytorch_model.bin", repo_id="adwod/Streamlite_ViT_2000")
13
  # Descargar el archivo de pesos del modelo y cargarlo en el modelo
14
  model.load_state_dict(torch.load(cached_download(model_path)))
15
  model = ViTForImageClassification(config)