Update app.py

app.py

@@ -1,24 +1,41 @@
 import gradio as gr
 import pickle
 
-# Load model
+# Load all models
+with open("knn_model.pkl", "rb") as f:
+    knn_model = pickle.load(f)
+
+with open("rf_model.pkl", "rb") as f:
+    rf_model = pickle.load(f)
+
 with open("svm_model.pkl", "rb") as f:
-    model = pickle.load(f)
+    svm_model = pickle.load(f)
+
+# Map model name to actual model
+model_map = {
+    "KNN": knn_model,
+    "Random Forest": rf_model,
+    "SVM": svm_model
+}
 
 # Prediction function
-def predict(sex, pregnant, TT4, T3, T4U, FTI, TSH):
+def predict(model_name, sex, pregnant, on_thyroxine, TT4, T3, T4U, FTI, TSH):
     try:
-        # Convert inputs to appropriate types
+        model = model_map[model_name]
+
+        # Ensure inputs are correctly typed
         sex = int(sex)
         pregnant = int(pregnant)
+        on_thyroxine = int(on_thyroxine)
         TT4 = float(TT4)
         T3 = float(T3)
         T4U = float(T4U)
         FTI = float(FTI)
         TSH = float(TSH)
-        prediction = model.predict([[sex, pregnant, TT4, T3, T4U, FTI, TSH]])
+
+        prediction = model.predict([[sex, pregnant, on_thyroxine, TT4, T3, T4U, FTI, TSH]])
         label_map = {0: "Hyperthyroid", 1: "Hypothyroid", 2: "Negative"}
-        return f"Prediction: {label_map.get(prediction[0], 'Unknown')}"
+        return f"Prediction using {model_name}: {label_map.get(prediction[0], 'Unknown')}"
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -26,8 +43,10 @@ def predict(sex, pregnant, TT4, T3, T4U, FTI, TSH):
 demo = gr.Interface(
     fn=predict,
     inputs=[
+        gr.Dropdown(["SVM", "KNN", "Random Forest"], label="Select Model"),
         gr.Radio([0, 1], label="Sex (0: Female, 1: Male)"),
         gr.Radio([0, 1], label="Pregnant (0: No, 1: Yes)"),
+        gr.Radio([0, 1], label="On Thyroxine (0: No, 1: Yes)"),
         gr.Number(label="TT4"),
         gr.Number(label="T3"),
         gr.Number(label="T4U"),
@@ -35,8 +54,8 @@ demo = gr.Interface(
         gr.Number(label="TSH"),
     ],
     outputs="text",
-    title="Hyperthyroid Prediction",
-    description="Enter patient info to predict thyroid condition using SVM model."
+    title="Hyperthyroid Prediction (Multi-Model)",
+    description="Select a model and input medical features to predict thyroid condition."
 )
 
 if __name__ == "__main__":