fixed errors

app.py

@@ -42,7 +42,7 @@ def run_prediction(image: gr.Image, selected_model: str)-> tuple[torch.Tensor]:
 with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
     gr.Markdown("## Semantic Segmentation with Real-Time Networks")
     gr.Markdown('A small user interface created to run semantic segmentation on images using city scapes like predictions and real time segmentation networks.')
-    gr.Markdown("Upload an image and choose your preferred model for segmentation.")
+    gr.Markdown("Upload an image and choose your preferred model for segmentation, or otherwise use one of the preloaded images.")
 
     with gr.Row():
         with gr.Column():
@@ -54,7 +54,7 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
             image_input = gr.Image(type="pil", label="Upload image")
             submit_btn = gr.Button("Run prediction")
         with gr.Column():
-            result_display = gr.Image(label="Model prediction", visible=False)
+            result_display = gr.Image(label="Model prediction", visible=True)
             error_text = gr.Markdown("", visible=False)
 
     submit_btn.click(

utils/imageHandling.py

@@ -63,10 +63,10 @@ def print_mask(mask:torch.Tensor, numClasses:int=19)->None:
         (119, 11, 32)    # 18: bicycle
     ]
 
-    new_mask = torch.zeros((mask.shape[0], mask.shape[1], 3),dtype=torch.uint8)
-    new_mask[mask == 255] = (0,0,0)
+    new_mask = torch.zeros((mask.shape[0], mask.shape[1], 3), dtype=torch.uint8)
+    new_mask[mask == 255] = torch.tensor([0, 0, 0], dtype=torch.uint8)
     for i in range (numClasses):
-        new_mask[mask == i] = colors[i][:3]
+        new_mask[mask == i] = torch.tensor(colors[i][:3], dtype=torch.uint8)
     return new_mask.permute(2,0,1)  
 
 # %% postprocessing