Spaces:
Paused
Paused
fix image processing mistake
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ def run_image_mixer(args):
|
|
118 |
|
119 |
# Clear GPU memory cache so less likely to OOM
|
120 |
torch.cuda.empty_cache()
|
121 |
-
return ims
|
122 |
|
123 |
def is_female(img):
|
124 |
pred,pred_idx,probs = gender_learn.predict(img)
|
@@ -152,7 +152,7 @@ def boutsify(person):
|
|
152 |
gradio_interface = gradio.Interface(
|
153 |
fn=boutsify,
|
154 |
inputs="image",
|
155 |
-
outputs=
|
156 |
title="Boutsify images",
|
157 |
description="Turn portraits into a painting in the style of Flemish master Dirck Bouts",
|
158 |
article="© iO Digital"
|
|
|
118 |
|
119 |
# Clear GPU memory cache so less likely to OOM
|
120 |
torch.cuda.empty_cache()
|
121 |
+
return ims[0]
|
122 |
|
123 |
def is_female(img):
|
124 |
pred,pred_idx,probs = gender_learn.predict(img)
|
|
|
152 |
gradio_interface = gradio.Interface(
|
153 |
fn=boutsify,
|
154 |
inputs="image",
|
155 |
+
outputs="image",
|
156 |
title="Boutsify images",
|
157 |
description="Turn portraits into a painting in the style of Flemish master Dirck Bouts",
|
158 |
article="© iO Digital"
|