Update app.py

app.py

@@ -28,7 +28,7 @@ class MultiTaskModel(nn.Module):
 # 2. Reconstruct the Model and Load Weights
 ########################################
 # Set the number of object classes (update this to match your training)
-num_obj_classes = 139  # Example value; change it to your actual number
+num_obj_classes = 139  # Example value; update as needed
 
 device = torch.device("cpu")
 
@@ -43,7 +43,7 @@ model.to(device)
 
 # Download the state dict from HF Hub.
 repo_id = "Abdu07/multitask-model"  # Your repo name
-filename = "multitask_model_weights.pth"  # The state dict file you uploaded
+filename = "Yolloplusclassproject_weights.pth"  # New weight file name
 weights_path = hf_hub_download(repo_id=repo_id, filename=filename)
 
 # Load the state dict and update the model.