Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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()
|