Update app.py

app.py

@@ -211,8 +211,12 @@ model = YOLO()
 # model = torch.load("Model_IV.pt")
 # model.eval()
 checkpoint = torch.load("Model_IV.pt")
-model.load_state_dict(checkpoint)  # Load the saved weights
-model.eval()  # Set the model to evaluation mode
+# model.load_state_dict(checkpoint)  # Load the saved weights
+# model.eval()  # Set the model to evaluation mode
+
+print(type(checkpoint))
+if isinstance(checkpoint, dict):
+    print(checkpoint.keys())
 
 # Define preprocessing
 transform = T.Compose([