trying to figure out state dict

app.py

@@ -296,9 +296,9 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     # print(model_url)
     
     loaded = torch.load(model_url, map_location='cpu', )
-    # print(loaded)
+    print(loaded.keys())
     
-    model.load_state_dict(loaded)
+    model.load_state_dict(loaded['state_dict'])
     # print(model)
     
     # model = EfficientNetPreTrained(config)