dd

app.py

@@ -5,6 +5,11 @@ import torch
 import torchvision
 
 model = torch.load('v4-epoch=19-val_loss=0.6964-val_accuracy=0.8964.ckpt', map_location=torch.device('cpu'))
+state_dict = checkpoint["state_dict"]                                                                                                                                                                                                         
+model_weights = state_dict                                                                                                                                                                                                                    
+for key in list(model_weights):                                                                                                                                                                                                               
+    model_weights[key.replace("backbone.", "")] = model_weights.pop(key)                                                                                                                                                                      
+model.load_state_dict(model_weights).eval()
 
 import requests
 from PIL import Image