tombio commited on
Commit
420f037
1 Parent(s): 88e8a01

generate three images

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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[0]
122
 
123
  def is_female(img):
124
  pred,pred_idx,probs = gender_learn.predict(img)
@@ -144,7 +144,7 @@ def boutsify(person):
144
  person_image,
145
  Image.open("bouts_m24.jpg").convert("RGB"),
146
  1.2,1,1,1.4,1,
147
- 3.0, 1, random.randrange(0, 10000), 40,
148
  ]
149
 
150
  return run_image_mixer(inputs)
@@ -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
122
 
123
  def is_female(img):
124
  pred,pred_idx,probs = gender_learn.predict(img)
 
144
  person_image,
145
  Image.open("bouts_m24.jpg").convert("RGB"),
146
  1.2,1,1,1.4,1,
147
+ 3.0, 3, random.randrange(0, 10000), 40,
148
  ]
149
 
150
  return run_image_mixer(inputs)
 
152
  gradio_interface = gradio.Interface(
153
  fn=boutsify,
154
  inputs="image",
155
+ outputs=["image","image","image"],
156
  title="Boutsify images",
157
  description="Turn portraits into a painting in the style of Flemish master Dirck Bouts",
158
  article="© iO Digital"