-

app.py

@@ -10,9 +10,7 @@ def seg(image):
     inputs = feature_extractor(images=image, return_tensors="pt")
     outputs = model(**inputs)
     logits = outputs.logits
-    predicted_class_idx = logits.argmax(-1).item()
-    return model.config.id2label[predicted_class_idx]
-
+    return logits
     
 iface = gr.Interface(fn=seg, inputs=gr.inputs.Image(type='pil'), outputs='label')
 iface.launch()