Spaces:
Build error
Build error
Akash James
commited on
Commit
·
c9fd122
1
Parent(s):
1c5ae49
Default CUDA and Post Processing as True; Potential CPU fix
Browse files
app.py
CHANGED
|
@@ -24,6 +24,8 @@ def run(content_img, style_img, cuda, post_processing, fast):
|
|
| 24 |
|
| 25 |
if cuda:
|
| 26 |
p_wct.cuda(0)
|
|
|
|
|
|
|
| 27 |
|
| 28 |
output_img = process_stylization.stylization_gradio(
|
| 29 |
stylization_module=p_wct,
|
|
@@ -43,8 +45,8 @@ if __name__ == '__main__':
|
|
| 43 |
inputs=[
|
| 44 |
gr.Image(label='Content Image'),
|
| 45 |
gr.Image(label='Stylize Image'),
|
| 46 |
-
gr.Checkbox(value=
|
| 47 |
-
gr.Checkbox(value=
|
| 48 |
gr.Radio(choices=["Guided Image Filtering (Fast)", "Photorealisitic Smoothing (Slow)"], value="Guided Image Filtering (Fast)", type="index", label="Algorithm", interactive=False),
|
| 49 |
],
|
| 50 |
outputs=[gr.Image(
|
|
|
|
| 24 |
|
| 25 |
if cuda:
|
| 26 |
p_wct.cuda(0)
|
| 27 |
+
else:
|
| 28 |
+
p_wct.to('cpu')
|
| 29 |
|
| 30 |
output_img = process_stylization.stylization_gradio(
|
| 31 |
stylization_module=p_wct,
|
|
|
|
| 45 |
inputs=[
|
| 46 |
gr.Image(label='Content Image'),
|
| 47 |
gr.Image(label='Stylize Image'),
|
| 48 |
+
gr.Checkbox(value=True, label='Use CUDA'),
|
| 49 |
+
gr.Checkbox(value=True, label='Post Processing'),
|
| 50 |
gr.Radio(choices=["Guided Image Filtering (Fast)", "Photorealisitic Smoothing (Slow)"], value="Guided Image Filtering (Fast)", type="index", label="Algorithm", interactive=False),
|
| 51 |
],
|
| 52 |
outputs=[gr.Image(
|