Update app.py

app.py

@@ -11,9 +11,10 @@ config = ViTConfig.from_json_dict(cached_download(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")
 # Descargar el archivo de pesos del modelo y cargarlo en el modelo
-model.load_state_dict(torch.load(cached_download(model_path)))
 model = ViTForImageClassification(config)
-model.load_state_dict(torch.load(model_path))
+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')