Update app.py

app.py

@@ -3,22 +3,19 @@ import torch
 from transformers import ViTFeatureExtractor, ViTForImageClassification, ViTConfig
 from huggingface_hub import hf_hub_url, cached_download
 
-# URL del archivo de configuración en el espacio de Hugging Face
-
-config_path = "~/Descargas/config.json"
-config = ViTConfig.from_pretrained(config_path)
-
-model_path = "~/Descargas/pytorch_model.bin"
-model.load_state_dict(torch.load(model_path))
+from huggingface_hub import hf_hub_url, cached_download
 
+# URL del archivo de configuración en el espacio de Hugging Face
+config_url = hf_hub_url(filename="config.json", repo_id="/adwod/Streamlite_ViT_2000/blob/main/config.json")
+# Descargar el archivo de configuración y cargarlo en una instancia de ViTConfig
+config = ViTConfig.from_pretrained(config_url)
 
 
 # URL del archivo de pesos del modelo en el espacio de Hugging Face
-
+model_path = hf_hub_url(filename="pytorch_model.bin", repo_id="/adwod/Streamlite_ViT_2000/blob/main/pytorch_model.bin")
 # Descargar el archivo de pesos del modelo y cargarlo en el modelo
 model = ViTForImageClassification(config)
-
-
+model.load_state_dict(torch.load(cached_download(model_path)))
 
 
 # Cargar el extractor de características