Update app.py

app.py

@@ -9,6 +9,7 @@ SECRET_TOKEN = os.getenv("SECRET_TOKEN")
 API_URL = "https://api-inference.huggingface.co/models/facebook/detr-resnet-50-dc5-panoptic"
 headers = {"Authorization": f'Bearer {SECRET_TOKEN}'}
 
+estimator = pipeline("depth-estimation")
 
 def rb(img):
     # initialiaze io to_bytes converter
@@ -17,14 +18,13 @@ def rb(img):
     img.save(img_byte_arr, format='JPEG', subsampling=0, quality=100)
     # converts image array to bytesarray
     img_byte_arr = img_byte_arr.getvalue()
-    return img_byte_arr
-
-estimator = pipeline("depth-estimation")
-result = estimator("http://images.cocodataset.org/val2017/000000039769.jpg")
+    result = estimator(img_byte_arr)
+    return result
 
 
 inputs = gr.inputs.Image(type="pil", label="Upload an image")
 
-demo = gr.Interface(fn=rb, inputs=inputs, outputs=result["depth"])
+
+demo = gr.Interface(fn=rb, inputs=inputs, outputs="json")
 #demo = gr.Interface(fn=rb, inputs=inputs, outputs=result["depth"])
 demo.launch()