tombio commited on
Commit
1b19198
1 Parent(s): 7322f39

convert from array to image

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -119,17 +119,20 @@ def run_image_mixer(args):
119
  torch.cuda.empty_cache()
120
  return ims[0]
121
 
122
- def guess_gender(img):
123
  pred,pred_idx,probs = gender_learn.predict(img)
124
- return {gender_labels[i]: float(probs[i]) for i in range(len(gender_labels))}
125
 
126
 
127
  import gradio
128
 
129
  def boutsify(person):
130
- print(person)
131
- print(guess_gender(person))
132
- print(person)
 
 
 
133
 
134
  inputs = [
135
  "Image", "Image", "Text/URL", "Image", "Nothing",
@@ -137,7 +140,7 @@ def boutsify(person):
137
  Image.open("ex2-1.jpeg").convert("RGB"),
138
  Image.open("ex2-2.jpeg").convert("RGB"),
139
  Image.open("blonder.jpeg").convert("RGB"),
140
- person,
141
  Image.open("blonder.jpeg").convert("RGB"),
142
  1,1,1.5,1.4,1,
143
  3.0, 1, 0, 40,
 
119
  torch.cuda.empty_cache()
120
  return ims[0]
121
 
122
+ def is_female(img):
123
  pred,pred_idx,probs = gender_learn.predict(img)
124
+ return float(probs[0]) > float(probs[1])
125
 
126
 
127
  import gradio
128
 
129
  def boutsify(person):
130
+ female_detected = is_female(person)
131
+
132
+ if female_detected:
133
+ print("Picture of a female")
134
+
135
+ person_image = Image.fromarray(person)
136
 
137
  inputs = [
138
  "Image", "Image", "Text/URL", "Image", "Nothing",
 
140
  Image.open("ex2-1.jpeg").convert("RGB"),
141
  Image.open("ex2-2.jpeg").convert("RGB"),
142
  Image.open("blonder.jpeg").convert("RGB"),
143
+ person_image,
144
  Image.open("blonder.jpeg").convert("RGB"),
145
  1,1,1.5,1.4,1,
146
  3.0, 1, 0, 40,