changed target dir

app.py

@@ -27,7 +27,7 @@ model = Model(vit, roberta, tokenizer, device).to(device)
 model.eval()
 
 # Initialize trained model
-state = torch.load('saved_model')
+state = torch.load('saved_model', map_location=torch.device('cpu'))
 model.load_state_dict(state['val_model_dict'])
 
 # Transform for input image