Update app.py

app.py

@@ -37,7 +37,7 @@ def load_model(model_dir: str, device=torch.device("cpu")):
     )
     
     # Load custom weights
-    state_dict = torch.load(os.path.join(model_dir, "model_state.bin"), map_location=device)
+    state_dict = torch.load(os.path.join(model_dir, "model_state.bin"))
     model.load_state_dict(state_dict)
     
     # Load tokenizer