Update app.py

app.py

@@ -25,8 +25,16 @@ import base64
 template_path = 'report_template_tables.html'
 dive_data = {}
 
-with open('./segmentation_error_data.pkl', 'rb') as f:
-    dive_data_precomputed = pickle.load(f)
+class CPU_Unpickler(pickle.Unpickler):
+    def find_class(self, module, name):
+        if module == 'torch.storage' and name == '_load_from_bytes':
+            return lambda b: torch.load(io.BytesIO(b), map_location='cpu')
+        else: return super().find_class(module, name)
+
+dive_data_precomputed = CPU_Unpickler(open('./segmentation_error_data.pkl', 'rb')).load()
+
+# with open('./segmentation_error_data.pkl', 'rb') as f:
+#     dive_data_precomputed = pickle.load(f)
 
 import sys
 import csv