nonsense

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,11 +30,10 @@ 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 = outputs.logits.detach().cpu()
+    logits = nn.functional.interpolate(outputs.logits.detach().cpu(),
+                    size=image.size[::-1], # (height, width)
+                    mode='bilinear',
+                    align_corners=False)
 
     # Second, apply argmax on the class dimension
     seg = logits.argmax(dim=1)[0]