Update app.py

app.py

@@ -19,6 +19,6 @@ def segment(image):
     predicted_label = logits.argmax(-1).item()
     # print(model.config.id2label[predicted_label])
     
-    return image #model.config.id2label[predicted_label]
+    return model.config.id2label[predicted_label]
 
-gr.Interface(fn=segment, inputs="image", outputs="image").launch()
+gr.Interface(fn=segment, inputs="image", outputs="text").launch()