Update app.py

app.py

@@ -10,7 +10,18 @@ from infer import *
 import logging
 from infer_tools.infer_tool import *
 map_location=torch.device('cpu')
+##EDIT FOR CPU
+# Open the file and read it into a string
+with open("/home/user/.local/lib/python3.8/site-packages/torch/serialization.py") as f:
+    text = f.read()
 
+# Replace the original line with the new line
+text = text.replace("map_location: MAP_LOCATION = None,", "map_location: MAP_LOCATION = 'CPU',")
+
+# Save the modified string to the original file
+with open("/home/user/.local/lib/python3.8/site-packages/torch/serialization.py", "w") as f:
+    f.write(text)
+############
 logging.getLogger('numba').setLevel(logging.WARNING)
 
 # 工程文件夹名，训练时用的那个