Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -164,7 +164,7 @@ gr.Interface(
164
  gr.inputs.Radio(["Robust ResNet50"], label="Backbone"),
165
  gr.inputs.Dropdown(["The Scream", "Rain Princess", "Starry Night", "The Mosaic"], type="value", default="Rain Princess", label="style")
166
  ],
167
- outputs = [gr.outputs.Image(label="Stylized"), gr.outputs.HTML(label="Comment")],
168
  # examples = [
169
  # []
170
  # ],
 
164
  gr.inputs.Radio(["Robust ResNet50"], label="Backbone"),
165
  gr.inputs.Dropdown(["The Scream", "Rain Princess", "Starry Night", "The Mosaic"], type="value", default="Rain Princess", label="style")
166
  ],
167
+ outputs = [gr.outputs.Image(type="pil", label="Stylized"), gr.outputs.HTML(label="Comment")],
168
  # examples = [
169
  # []
170
  # ],