Update app.py

app.py

@@ -14,9 +14,9 @@ def query(filename):
         data = f.read()
     response = requests.post(API_URL, headers=headers, data=data)
     return response.json()
-
-def rb(data):
-    response = requests.post(API_URL, headers=headers, data=data.tobytes())
+data = bytes(im.getdata())
+def rb(img):
+    response = requests.post(API_URL, headers=headers, data=bytes(img.getdata())
     return response.json()
 
 # train = os.listdir("./")
@@ -26,5 +26,5 @@ output = query("./09_truck.jpg")
 
 
 inputs = gr.inputs.Image(type="pil", label="Upload an image")
-demo = gr.Interface(fn=rb, inputs=inputs, outputs=output)
+demo = gr.Interface(fn=rb, inputs=inputs, outputs="json")
 demo.launch()