fix RuntimeError 'cpu'

app.py

@@ -1,6 +1,7 @@
 import torch
 from monai.bundle import ConfigParser
 import gradio as gr
+import json
 
 parser = ConfigParser() #  load configuration files that specify various parameters for running the MONAI workflow.
 parser.read_config(f="configs/inference.json") # read the config from specified JSON file
@@ -9,8 +10,9 @@ parser.read_meta(f="configs/metadata.json") # read the metadata from specified J
 inference = parser.get_parsed_content("inferer")
 network = parser.get_parsed_content("network_def")
 preprocess = parser.get_parsed_content("preprocessing")
-state_dict = torch.load("models/model.pt")
+state_dict = torch.load("models/model.pt", map_location=torch.device('cpu'))
 network.load_state_dict(state_dict, strict=True) #  Loads a model’s parameter dictionary
+
 class_names = {
     0: "Other",
     1: "Inflammatory",
@@ -21,6 +23,7 @@ class_names = {
 def classify_image(image_file, label_file):
     data = {"image":image_file, "label":label_file}
     batch = preprocess(data)
+    batch['image'] = batch['image']
     network.eval()
     with torch.no_grad():
         pred = inference(batch['image'].unsqueeze(dim=0), network) # expect 4 channels input  (3 RGB, 1 Label mask)
@@ -29,25 +32,25 @@ def classify_image(image_file, label_file):
     return confidences
 
 example_files1 = [
-    [r'sample_data\Images\test_11_2_0628.png',
-    r'sample_data\Labels\test_11_2_0628.png'],
-    [r'sample_data\Images\test_9_4_0149.png',
-    r'sample_data\Labels\test_9_4_0149.png'],
-    [r'sample_data\Images\test_12_3_0292.png',
-    r'sample_data\Labels\test_12_3_0292.png'],
-    [r'sample_data\Images\test_9_4_0019.png',
-    r'sample_data\Labels\test_9_4_0019.png']
+    ['sample_data/Images/test_11_2_0628.png',
+    'sample_data/Labels/test_11_2_0628.png'],
+    ['sample_data/Images/test_9_4_0149.png',
+    'sample_data/Labels/test_9_4_0149.png'],
+    ['sample_data/Images/test_12_3_0292.png',
+    'sample_data/Labels/test_12_3_0292.png'],
+    ['sample_data/Images/test_9_4_0019.png',
+    'sample_data/Labels/test_9_4_0019.png']
 ]
 
 example_files2 = [
-    [r'sample_data\Images\test_14_3_0433.png',
-    r'sample_data\Labels\test_14_3_0433.png'],
-    [r'sample_data\Images\test_14_4_0544.png',
-    r'sample_data\Labels\test_14_4_0544.png'],
-    [r'sample_data\Images\train_1_1_0095.png',
-    r'sample_data\Labels\train_1_1_0095.png'],
-    [r'sample_data\Images\train_1_3_0020.png',
-    r'sample_data\Labels\train_1_3_0020.png'],
+    ['sample_data/Images/test_14_3_0433.png',
+    'sample_data/Labels/test_14_3_0433.png'],
+    ['sample_data/Images/test_14_4_0544.png',
+    'sample_data/Labels/test_14_4_0544.png'],
+    ['sample_data/Images/train_1_1_0095.png',
+    'sample_data/Labels/train_1_1_0095.png'],
+    ['sample_data/Images/train_1_3_0020.png',
+    'sample_data/Labels/train_1_3_0020.png'],
 ]
 
 with open('Description.md','r') as file: