Update app.py

app.py

@@ -5,19 +5,21 @@ from huggingface_hub import hf_hub_url, cached_download
 
 from huggingface_hub import hf_hub_url, cached_download
 
+# URL del repositorio en Hugging Face
+repo_id = "adwod/Streamlite_ViT_2000"
+
 # 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")
+config_url = hf_hub_url(filename="config.json", repo_id=repo_id)
 # 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")
+model_path = hf_hub_url(filename="pytorch_model.bin", repo_id=repo_id)
 # 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
 feature_extractor = ViTFeatureExtractor.from_pretrained('google/vit-base-patch16-224')