Update app.py

app.py

@@ -15,7 +15,7 @@ def query_image(inputs):
     response = requests.post(API_URL, headers=headers, json=payload)
     image_bytes = response.content
     image = Image.open(io.BytesIO(image_bytes))
-    return image
+    return image, inputs
 
 # Define Gradio Blocks UI
 with gr.Blocks(theme="nevreal/blues") as demo: