Update app.py

app.py

@@ -84,11 +84,12 @@ def check_image_format(filepath):
         return False
 
 # Function to predict hemoglobin value
+# Updated predict_hemoglobin function
 def predict_hemoglobin(age, gender, image):
     try:
-        # Check if the image file is valid
-        if not check_image_format(image.name):
-            return "Error: The uploaded image file is not recognized or is corrupt. Please upload an image in JPG, PNG, BMP, or GIF format."
+        # Ensure the image is not None
+        if image is None:
+            return "Error: No image uploaded. Please upload an image."
 
         # Extract features from the image
         features = extract_features(image)
@@ -124,23 +125,3 @@ def predict_hemoglobin(age, gender, image):
     except Exception as e:
         print(f"An error occurred during prediction: {e}")
         return "An error occurred during prediction. Please try again."
-
-# Gradio Interface setup
-def create_gradio_interface():
-    # Define the inputs and outputs for the Gradio interface
-    image_input = gr.Image(type="pil", label="Image (Upload Image)", interactive=True)
-    age_input = gr.Number(label="Age", value=25, precision=0)
-    gender_input = gr.Radio(choices=["Male", "Female"], label="Gender", value="Male")
-
-    # Set up the Gradio interface with the prediction function
-    iface = gr.Interface(fn=predict_hemoglobin,
-                         inputs=[age_input, gender_input, image_input],
-                         outputs="text",
-                         live=True,
-                         title="Hemoglobin Prediction")
-
-    iface.launch(share=True)  # Set share=True to create a public link
-
-# Run the Gradio app
-if __name__ == "__main__":
-    create_gradio_interface()