maybe this will fix the bug?

app.py

@@ -2,7 +2,7 @@ from matplotlib.pyplot import axis
 import gradio as gr
 import requests
 import numpy as np
-from torch import nn
+# from torch import nn
 from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 import requests
 
@@ -30,10 +30,11 @@ def inference(image):
     outputs = model(**inputs)
 
     # First, rescale logits to original image size
-    logits = nn.functional.interpolate(outputs.logits.detach().cpu(),
-                    size=image.size[::-1], # (height, width)
-                    mode='bilinear',
-                    align_corners=False)
+    # logits = nn.functional.interpolate(outputs.logits.detach().cpu(),
+    #                 size=image.size[::-1], # (height, width)
+    #                 mode='bilinear',
+    #                 align_corners=False)
+    logits = outputs.logits.detach().cpu()
 
     # Second, apply argmax on the class dimension
     seg = logits.argmax(dim=1)[0]