Update app.py

app.py

@@ -6,13 +6,13 @@ from huggingface_hub import hf_hub_url, cached_download
 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")
+config_url = hf_hub_url(filename="config.json", repo_id="/adwod/Streamlite_ViT_2000")
 # 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")
+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)))