Update app.py

app.py

@@ -81,7 +81,13 @@ def create_gradio_app(api_spec, api_url):
     inputs.append(data_field)
     print(output_schema)
     if output_schema["type"] == "string":
-        output_component = gr.Textbox(label="Output")
+        if "format" in output_schema:
+            if(output_schema["format"] == "uri"):
+                output_component = gr.Gallery(label=output_schema["title"])
+            else:
+                output_component = gr.Textbox(label="Output")
+        else:
+            output_component = gr.Textbox(label="Output")
         outputs.append(output_component)
     elif output_schema["type"] == "array":
         if "format" in output_schema["items"]:
@@ -101,7 +107,7 @@ def create_gradio_app(api_spec, api_url):
         payload = {"input": {}}
         for i, key in enumerate(keys):
             value = args[i]
-            if(os.path.exists(value)):
+            if value and (os.path.exists(str(value))):
                 value = "http://localhost:7860/file=" + value
             payload["input"][key] = value
         print(payload)