Update app.py

app.py

@@ -1,16 +1,36 @@
+import gradio as gr
 import requests
-import os
+from PIL import Image
+
 
-WD = os.getcwd()
-print(WD)
 
 API_URL = "https://api-inference.huggingface.co/models/facebook/detr-resnet-50-panoptic"
 headers = {"Authorization": "Bearer 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"}
 
-def query(filename):
-    with open(filename.name, "rb") as f:
+
+inputs = gr.inputs.Image(type="pil", label="Upload an image")
+# output = query("cats.jpg")
+
+
+# Perform image segmentation for multy class output
+def query(inputs):
+    with open(inputs, "rb") as f:
         data = f.read()
     response = requests.post(API_URL, headers=headers, data=data)
     return response.json()
 
-output = query("04_deer.jpg")
+
+inputs = gr.inputs.Image(type="pil", label="Upload an image")
+# outputs = gr.outputs.HTML() #uncomment for single class output 
+#outputs = query(inputs)
+
+title = "<h1 style='text-align: center;'>Image Segmentation</h1>"
+description = "Upload an image and get the segmentation result."
+
+gr.Interface(fn=classify_image, 
+             inputs=inputs, 
+             outputs=query(inputs.get_name()), 
+             title=title, 
+             examples=[["00_plane.jpg"], ["01_car.jpg"], ["02_bird.jpg"], ["03_cat.jpg"], ["04_deer.jpg"]],
+             description=description,
+             enable_queue=True).launch()