Update app.py

app.py

@@ -83,7 +83,7 @@ def create_gradio_app(api_spec, api_url):
     if output_schema["type"] == "string":
         if "format" in output_schema:
             if(output_schema["format"] == "uri"):
-                output_component = gr.Gallery(label=output_schema["title"])
+                output_component = gr.Image(label=output_schema["title"])
             else:
                 output_component = gr.Textbox(label="Output")
         else:
@@ -92,7 +92,7 @@ def create_gradio_app(api_spec, api_url):
     elif output_schema["type"] == "array":
         if "format" in output_schema["items"]:
             if(output_schema["items"]["format"] == "uri"):
-                output_component = gr.Gallery(label=output_schema["title"])
+                output_component = gr.Image(label=output_schema["title"])
             else:
                 output_component = gr.Textbox(label=output_schema["title"])
         else:
@@ -126,7 +126,7 @@ def create_gradio_app(api_spec, api_url):
                 image_stream = io.BytesIO(image_data)
                 output_images.append(Image.open(image_stream))
 
-            return output_images
+            return output_images[0], keys
         else:
             raise gr.Error("The submission failed!")
     return gr.Interface(fn=predict, inputs=inputs, outputs=outputs if outputs else "textbox")