Update prediction.py

prediction.py

@@ -77,7 +77,7 @@ def load_model():
 
     # Load the state_dict (weights) from the saved model file
     try:
-        state_dict = torch.load("transformer_model.bin",map_location=device) # Ensure loading on the correct device
+        state_dict = torch.load("transformer_model(1).bin",map_location=device) # Ensure loading on the correct device
         model.load_state_dict(state_dict)
         model.eval()
         model.to(device)  # Send model to GPU if available