InstructIR-API

Paused

Nymbo commited on Oct 1

Commit

eeb4bf1

verified ·

1 Parent(s): 349633f

Keep the /predict endpoint name; PNG output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,8 @@ demo = gr.Interface(
             gr.Image(type="pil", label="Input"),
             gr.Text(label="Prompt")
     ],
-    outputs=[gr.Image(type="pil", label="Ouput")],
     title=title,
     description=description,
     article=article,

             gr.Image(type="pil", label="Input"),
             gr.Text(label="Prompt")
     ],
+    outputs=[gr.Image(type="pil", label="Ouput", format="png")],
+    api_name="predict",
     title=title,
     description=description,
     article=article,