Update model.py

model.py

@@ -150,7 +150,7 @@ class Trainer:
     
     def predict(self, inputs):
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        self.model.load_state_dict(torch.load('best_model.pth', weights_only=True, map_location=torch.device('cpu'))
+        self.model.load_state_dict(torch.load('best_model.pth', weights_only=True, map_location=torch.device('cpu')))
         self.model.to(device)
         self.model.eval()
         inputs = inputs.to(device)