No model

src/app.py

@@ -25,7 +25,7 @@ vocab = [
 
 
 model = get_model()
-state = torch.load('exported_model.pth', map_location="cpu")["model"]
+state = torch.load('exported_model.pth', map_location="cpu")
 apply_weights(model, state, copy_weight)
 
 to_tensor = ToTensor()