Spaces:

yakhoub
/

PBO

Runtime error

yakhoub commited on Oct 28, 2024

Commit

b64ef75

verified ·

1 Parent(s): d19b416

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,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('ocr_model_large_2024-07-25_15_32.pt', map_location=torch.device('cpu')))
 trained_model.to(device)
 trained_model.eval()

 # 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('ocr_model_large_2024-08-28_14_44.pt', map_location=torch.device('cpu')))
 trained_model.to(device)
 trained_model.eval()