Fix: Version issue

app.py

@@ -17,8 +17,8 @@ def load_model(model_path: str):
     """
     Load the model.
     """
-    # Load the pre-trained ResNet50 model from ImageNet
-    model = models.resnet50(pretrained=False)
+    # Load the pre-trained ResNet50 model using the new weights parameter
+    model = models.resnet50(weights=None)
 
     # Load custom weights from a .pth file with CPU mapping
     state_dict = torch.load(model_path, map_location=torch.device('cpu'))
@@ -37,7 +37,8 @@ def load_classes():
     Load the classes.
     """
     # Get ImageNet class names from ResNet50 weights
-    classes = models.ResNet50_Weights.IMAGENET1K_V2.meta["categories"]
+    weights = models.ResNet50_Weights.IMAGENET1K_V2
+    classes = weights.meta["categories"]
     return classes
 
 
@@ -112,7 +113,7 @@ def main():
             )
 
         # Launch the demo
-        demo.launch(server_name="0.0.0.0", debug=True)
+        demo.launch(server_name="0.0.0.0", debug=True, ssr=False)
 
 
 if __name__ == "__main__":