Update app.py

app.py

@@ -28,23 +28,37 @@ def predict_gender(image):
     return "Male" if prediction[0] < 0.5 else "Female"
 
 
-def predict(image_in_video):
-    if image_in_video == None:
-        raise gr.Error("Please capture an image using the webcam.")
-    return image_in_video
+def predict(image_in_video, image_in_img):
+    if image_in_video == None and image_in_img == None:
+        raise gr.Error("Please capture an image using the webcam or upload an image.")
+    image = image_in_video or image_in_img
+    return image
 
 with gr.Blocks() as blocks:
-    gr.Markdown("### Capture Image Using WebCam") 
+    gr.Markdown("### Capture Image Using WebCam or Upload") 
 
     with gr.Row():
         with gr.Column():
+            image_or_file_opt = gr.Radio(["webcam", "file"], value="webcam",
+                                         label="How would you like to upload your image?")
             image_in_video = gr.Image(source="webcam", type="filepath")
+            image_in_img = gr.Image(source="upload", visible=False, type="filepath")
+
+            # Update visibility based on selection
+            def toggle(choice):
+                if choice == "webcam":
+                    return gr.update(visible=True, value=None), gr.update(visible=False, value=None)
+                else:
+                    return gr.update(visible=False, value=None), gr.update(visible=True, value=None)
+
+            image_or_file_opt.change(fn=toggle, inputs=[image_or_file_opt],
+                                     outputs=[image_in_video, image_in_img], queue=False, show_progress=False)
         with gr.Column():
             image_out = gr.Image()
-            
+
     run_btn = gr.Button("Run")
-    run_btn.click(fn=predict, inputs=[image_in_video], outputs=[image_out])
-    gr.Examples(fn=predict, examples=[], inputs=[image_in_video], outputs=[image_out])
+    run_btn.click(fn=predict, inputs=[image_in_img, image_in_video], outputs=[image_out])
+    gr.Examples(fn=predict, examples=[], inputs=[image_in_img, image_in_video], outputs=[image_out])
 
 blocks.queue()
 blocks.launch()