Update app.py

app.py

@@ -34,7 +34,7 @@ class ModelConfig:
 # Charger le modèle entraîné à partir du fichier .pt
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 trained_model = VisionEncoderDecoderModel.from_pretrained(ModelConfig.MODEL_NAME)
-trained_model.load_state_dict(torch.load('/https://huggingface.co/spaces/yakhoub/PBO/blob/main/ocr_model_large_2024-07-25_15_32.pt'))
+trained_model.load_state_dict(torch.load('ocr_model_large_2024-07-25_15_32.pt'))
 trained_model.to(device)
 trained_model.eval()