Spaces:
Paused
Paused
convert from array to image
Browse files
app.py
CHANGED
@@ -119,17 +119,20 @@ def run_image_mixer(args):
|
|
119 |
torch.cuda.empty_cache()
|
120 |
return ims[0]
|
121 |
|
122 |
-
def
|
123 |
pred,pred_idx,probs = gender_learn.predict(img)
|
124 |
-
return
|
125 |
|
126 |
|
127 |
import gradio
|
128 |
|
129 |
def boutsify(person):
|
130 |
-
|
131 |
-
|
132 |
-
|
|
|
|
|
|
|
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 |
-
|
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,
|