Update app.py

app.py

@@ -4,17 +4,20 @@ from transformers import ViTFeatureExtractor, ViTForImageClassification, ViTConf
 from huggingface_hub import hf_hub_url, cached_download
 
 # URL del archivo de configuración en el espacio de Hugging Face
-config_url = "https://huggingface.co/spaces/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)
+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))
+
 
 
 # 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")
+
 # 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)))
+