Adding checkbox for male and female inputs

predict.py

@@ -27,16 +27,21 @@ transform = A.Compose([
 
 
 def predict(image, gender):
+    if gender:
+        is_female = 1
+    else:
+        is_female = 0
+
     model = initialize_model()
 
     processed_image = transform(image=np.array(image, dtype=np.uint8))['image']
     processed_image = processed_image.unsqueeze(0)
     processed_image = processed_image.to(device)
-    gender = torch.tensor(int(gender)).unsqueeze(0).unsqueeze(1).to(device)
+    is_female = torch.tensor(is_female).unsqueeze(0).unsqueeze(1).to(device)
 
     scans = {
         'image': processed_image,
-        'gender': gender
+        'gender': is_female
     }
     preds = model(scans)
     return int(preds)
@@ -47,14 +52,9 @@ def run():
     gender_input = gr.Checkbox(label="Is Female?", info="Is the scan of a female?", default=True),
     output = gr.outputs.Textbox(label="Predicted Age")
 
-    if gender_input:
-        is_female = 1
-    else:
-        is_female = 0
-
     BAE = gr.Interface(
         fn=predict,
-        inputs=[image_input, is_female],
+        inputs=[image_input, gender_input],
         outputs=output,
     )