tombio commited on
Commit
6e1757f
1 Parent(s): d2035fb

fix image processing mistake

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=["image"],
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"