added segformer inference

run.py

@@ -2,8 +2,15 @@ import gradio as gr
 import numpy as np
 from time import sleep
 import torch
-from transformers import SegformerImageProcessor
-
+from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
+
+id2label = {0: 'seal', 255: 'bck'}
+label2id = {'seal': 0, 'bck': 255}
+model = SegformerForSemanticSegmentation.from_pretrained("nvidia/mit-b0",
+                                                         num_labels=2,
+                                                         id2label=id2label,
+                                                         label2id=label2id,
+)
 image_processor = SegformerImageProcessor(reduce_labels=True)
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")