Update app.py

app.py

@@ -84,8 +84,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print("Using device:", device)
 
 UNET_PATH = "models/unet.pth"               
-MODEL_BACT_PATH = "models/models_bacterial.pt"
-MODEL_VIRAL_PATH = "models/models_viral.pt"
+MODEL_BACT_PATH = "models/model_bacterial.pt"
+MODEL_VIRAL_PATH = "models/model_viral.pt"
 
 
 unet = UNet(in_channels=3, num_classes=1).to(device)