Returned model to cpu after the execution of each gpu function

app.py

@@ -156,6 +156,7 @@ def build_model_and_transforms(args):
     model.load_state_dict(checkpoint, strict=False)
 
     model.eval()
+    model.cpu()
 
     return model, data_transform