Update app.py

app.py

@@ -42,7 +42,7 @@ model.to(device)
 
 # Download the state dict from HF Hub
 repo_id = "Abdu07/multitask-model"
-filename = "best_model.pt"  # Make sure this is the state dict file
+filename = "best_model_new.pt"  # Make sure this is the state dict file
 model_path = hf_hub_download(repo_id=repo_id, filename=filename)
 state_dict = torch.load(model_path, map_location="cpu")
 model.load_state_dict(state_dict)