maze commited on
Commit
8b1d9c1
β€’
1 Parent(s): 31e6d73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -151,16 +151,17 @@ def main(image, backbone, style):
151
  isize = image.size
152
  image = process(image, transformer)
153
  s = f"The output image {str(image.size)} is processed by {backbone} based on input image {str(isize)}. <br> Please <b>rate</b> the generated image through the <b>Flag</b> button below!"
 
154
  return image, s
155
 
156
-
157
  gr.Interface(
158
  title = "Stylize",
159
  description = "Image generated based on Fast Style Transfer",
160
  fn = main,
161
  inputs = [
162
  gr.inputs.Image(),
163
- gr.inputs.Radio(["VGG19", "Robust ResNet50", "Standard ResNet50"], label="Backbone"),
164
  gr.inputs.Dropdown(["The Scream", "Rain Princess", "Starry Night", "The Mosaic"], type="value", default="Rain Princess", label="style")
165
  ],
166
  outputs = [gr.outputs.Image(label="Stylized"), gr.outputs.HTML(label="Comment")],
 
151
  isize = image.size
152
  image = process(image, transformer)
153
  s = f"The output image {str(image.size)} is processed by {backbone} based on input image {str(isize)}. <br> Please <b>rate</b> the generated image through the <b>Flag</b> button below!"
154
+ print(s)
155
  return image, s
156
 
157
+ # "Standard ResNet50", "VGG19"
158
  gr.Interface(
159
  title = "Stylize",
160
  description = "Image generated based on Fast Style Transfer",
161
  fn = main,
162
  inputs = [
163
  gr.inputs.Image(),
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")],