muneebable commited on
Commit
f89919c
·
verified ·
1 Parent(s): ad59550

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -173,14 +173,6 @@ def style_transfer(content_image, style_image, alpha, beta, conv1_1, conv2_1, co
173
 
174
 
175
  # Example images
176
- # examples = [
177
- # ["https://huggingface.co/spaces/muneebable/vgg-style-transfer/resolve/main/assets/content_1.jpg",
178
- # "https://huggingface.co/spaces/muneebable/vgg-style-transfer/resolve/main/assets/style_1.jpg"],
179
- # ["https://huggingface.co/spaces/muneebable/vgg-style-transfer/resolve/main/assets/content_2.jpg",
180
- # "https://huggingface.co/spaces/muneebable/vgg-style-transfer/resolve/main/assets/style_2.jpg"],
181
- # ["https://huggingface.co/spaces/muneebable/vgg-style-transfer/resolve/main/assets/content_3.png",
182
- # "https://huggingface.co/spaces/muneebable/vgg-style-transfer/resolve/main/assets/style_3.jpg"],
183
- # ]
184
 
185
  examples = [
186
  ["assets/content_1.jpg",
@@ -191,6 +183,12 @@ examples = [
191
  "assets/style_3.jpg"],
192
  ]
193
 
 
 
 
 
 
 
194
  #Load VGG19 model
195
  vgg = models.vgg19(pretrained=True).features
196
  for param in vgg.parameters():
@@ -247,7 +245,8 @@ with gr.Blocks() as demo:
247
 
248
  gr.Examples(
249
  resized_examples,
250
- inputs=[content_input, style_input]
 
251
  )
252
 
253
  demo.launch()
 
173
 
174
 
175
  # Example images
 
 
 
 
 
 
 
 
176
 
177
  examples = [
178
  ["assets/content_1.jpg",
 
183
  "assets/style_3.jpg"],
184
  ]
185
 
186
+ examples_output = [
187
+ "assets/result_1.png",
188
+ "assets/result_2.png",
189
+ "assets/result_3.png"
190
+ ]
191
+
192
  #Load VGG19 model
193
  vgg = models.vgg19(pretrained=True).features
194
  for param in vgg.parameters():
 
245
 
246
  gr.Examples(
247
  resized_examples,
248
+ inputs=[content_input, style_input],
249
+ outputs=examples_output
250
  )
251
 
252
  demo.launch()