update app

app.py

@@ -111,7 +111,7 @@ if __name__ == '__main__':
     with open('config.yaml') as f:
         PARAMS = yaml.load(f, Loader=SafeLoader)
         print(PARAMS)
-    model = WeedModel.load_from_checkpoint("model/epoch=08.ckpt", params=PARAMS)
+    model = WeedModel.load_from_checkpoint("model/epoch=08.ckpt", params=PARAMS, map_location=torch.device('cpu'))
     model.eval()
 
     transform = transforms.Compose([