Update app.py

app.py

@@ -13,7 +13,7 @@ model = Conv5_FC3(input_size= [
         169,
         208,
         179
-    ])
+    ], gpu = False)
 model.load_state_dict(checkpoint_state["model"])
 model.eval()