Update app.py

app.py

@@ -0,0 +1,25 @@
+from transformers import pipeline
+from helper import load_image_from_url, render_results_in_image
+
+od_pipe = pipeline("object-detection", "facebook/detr-resnet-50")
+
+import gradio as gr
+
+def get_pipeline_prediction(pil_image):
+    #first get the pipeline output given the pil image
+    pipeline_output = od_pipe(pil_image)
+
+    #then process the image using the pipeline output
+    processed_image = render_results_in_image(pil_image, pipeline_output)
+    return processed_image
+
+demo = gr.Interface(
+    fn= get_pipeline_prediction,
+    inputs=gr.Image(label="Input Image",
+                    type="pil"),
+                    outputs=gr.Image(label="Output Image with predictions",
+                                     type="pil"),
+    title="Object Detection API",
+    description="Just upload your image and let ObjectDetect API work its magic, revealing the objects waiting to be discovered"
+)
+demo.launch()