Update app.py

app.py

@@ -1,60 +1,60 @@
-import gradio as gr
-import pandas as pd
-from pycaret.classification import load_model
-
-# Load the saved model
-model = load_model('best_automl_model')
-
-# Prediction function
-def predict_anemia(sex, red_pixel, green_pixel, blue_pixel):
-    # Map input values to a DataFrame
-    input_data = pd.DataFrame({
-        'Red Pixel': [red_pixel],
-        'Green Pixel': [green_pixel],
-        'Blue Pixel': [blue_pixel],
-        'Sex': [1 if sex == "Male" else 0]  # Encode 'Male' as 1, 'Female' as 0
-    })
-    
-    # Make predictions
-    prediction = model.predict(input_data)[0]
-    probability = model.predict_proba(input_data)[0][1]
-    
-    # Convert results
-    prediction_label = "Yes" if prediction == 1 else "No"
-    return prediction_label, f"{probability:.2%}"  # Probability as a percentage
-
-# Define input components with better labels and layout
-inputs = [
-    gr.Radio(["Male", "Female"], label="Sex", interactive=True),
-    gr.Slider(0, 100, step=0.1, label="Red Pixel Percentage (%)"),
-    gr.Slider(0, 100, step=0.1, label="Green Pixel Percentage (%)"),
-    gr.Slider(0, 100, step=0.1, label="Blue Pixel Percentage (%)"),
-]
-
-# Define output components
-outputs = [
-    gr.Textbox(label="Prediction", interactive=False),
-    gr.Textbox(label="Probability of Anemia (%)", interactive=False),
-]
-
-# Create the Gradio Interface
-interface = gr.Interface(
-    fn=predict_anemia,
-    inputs=inputs,
-    outputs=outputs,
-    title="✨ Anemia Prediction App ✨",
-    description=(
-        "This app predicts whether a person is anemic based on the percentages of red, "
-        "green, and blue pixels in their image and their sex. "
-        "Simply adjust the sliders and select the person's sex to get a prediction."
-    ),
-    theme="huggingface",  # Optional theme
-    live=False,  # Turn off live predictions for better performance
-    examples=[
-        ["Male", 45.0, 30.0, 25.0],
-        ["Female", 50.0, 28.0, 22.0],
-    ],
-)
-
-# Launch the app
-interface.launch()
+import gradio as gr
+import pandas as pd
+from pycaret.classification import load_model
+
+# Load the saved model
+model = load_model('best_automl_model')
+
+# Prediction function
+def predict_anemia(sex, red_pixel, green_pixel, blue_pixel):
+    # Map input values to a DataFrame
+    input_data = pd.DataFrame({
+        'Red Pixel': [red_pixel],
+        'Green Pixel': [green_pixel],
+        'Blue Pixel': [blue_pixel],
+        'Sex': [1 if sex == "Male" else 0]  # Encode 'Male' as 1, 'Female' as 0
+    })
+    
+    # Make predictions
+    prediction = model.predict(input_data)[0]
+    probability = model.predict_proba(input_data)[0][1]
+    
+    # Convert results
+    prediction_label = "Yes" if prediction == 1 else "No"
+    return prediction_label, f"{probability:.2%}"  # Probability as a percentage
+
+# Define input components with better labels and layout
+inputs = [
+    gr.Radio(["Male", "Female"], label="Sex", interactive=True),
+    gr.Slider(0, 100, step=0.1, label="Red Pixel Percentage (%)"),
+    gr.Slider(0, 100, step=0.1, label="Green Pixel Percentage (%)"),
+    gr.Slider(0, 100, step=0.1, label="Blue Pixel Percentage (%)"),
+]
+
+# Define output components
+outputs = [
+    gr.Textbox(label="Prediction", interactive=False),
+    gr.Textbox(label="Probability of Anemia (%)", interactive=False),
+]
+
+# Create the Gradio Interface
+interface = gr.Interface(
+    fn=predict_anemia,
+    inputs=inputs,
+    outputs=outputs,
+    title="Anemia Prediction App",
+    description=(
+        "This app predicts whether a person is anemic based on the percentages of red, "
+        "green, and blue pixels in their image and their sex. "
+        "Simply adjust the sliders and select the person's sex to get a prediction."
+    ),
+    theme="huggingface",  # Optional theme
+    live=False,  # Turn off live predictions for better performance
+    examples=[
+        ["Male", 45.0, 30.0, 25.0],
+        ["Female", 50.0, 28.0, 22.0],
+    ],
+)
+
+# Launch the app
+interface.launch()