Update app.py

app.py

@@ -137,4 +137,20 @@ def detect_obj(input_image):
         return output_pil_img
 
     except:
-        return input_image
+        return input_image
+
+
+with gr.Blocks(title="YOLOS Object Detection - ClassCat", css=".gradio-container {background:lightyellow;}") as demo:
+    gr.HTML('<h1>Yolo Object Detection</h1>')
+    gr.HTML("<h4>supported objects are [aeroplane,bicycle,bird,boat,bottle,bus,car,cat,chair,cow,diningtable,dog,horse,motorbike,person,pottedplant,sheep,sofa,train,tvmonitor]</h4>")
+    with gr.Row():
+        input_image = gr.Image(label="Input image", type="pil")
+        output_image = gr.Image(label="Output image", type="pil")
+
+    send_btn = gr.Button("Detect")
+    gr.Examples(['./samples/out_1.jpg'], inputs=input_image)
+
+    send_btn.click(fn=detect_obj, inputs=[input_image], outputs=[output_image])
+
+
+demo.launch(debug=True)