Update app.py

app.py

@@ -12,19 +12,13 @@ headers = {"Authorization": f'Bearer {SECRET_TOKEN}'}
 estimator = pipeline("depth-estimation")
 
 def rb(img):
-    # initialiaze io to_bytes converter
-    img_byte_arr = io.BytesIO()
-    # define quality of saved array
-    img.save(img_byte_arr, format='JPEG', subsampling=0, quality=100)
-    # converts image array to bytesarray
-    img_byte_arr = img_byte_arr.getvalue()
-    result = estimator(img_byte_arr)
+    result = estimator(img)
     return result
 
 
 inputs = gr.inputs.Image(type="pil", label="Upload an image")
 
 
-demo = gr.Interface(fn=rb, inputs=inputs, outputs="json")
+demo = gr.Interface(fn=estimator, inputs=inputs, outputs="image")
 #demo = gr.Interface(fn=rb, inputs=inputs, outputs=result["depth"])
 demo.launch()