Update app.py

app.py

@@ -493,13 +493,15 @@ from torchvision import transforms, utils
 from matplotlib import pyplot as plt
 import numpy as np
 
+model = SimpleCNN()
+model.load_state_dict(torch.load("model.pth"))
+model.eval()
 
 def predict_pose(test_image):
   img = cv2.resize(test_image, (32,32))
   convert_tensor = transforms.ToTensor()
   tensor_img = convert_tensor(img)
   tensor_img = tensor_img[None,:,:,:]
-  model.eval()
 
   outputs = model(tensor_img)