Update app.py

app.py

@@ -1,22 +1,14 @@
 import gradio as gr
-import joblib
+import pickle
 
-# Load models using joblib
-knn_model = joblib.load("knn_model.joblib")
-rf_model = joblib.load("rf_model.joblib")
-svm_model = joblib.load("svm_model.joblib")
-
-model_map = {
-    "KNN": knn_model,
-    "Random Forest": rf_model,
-    "SVM": svm_model
-}
+# Load the SVM model
+with open("svm_model.pkl", "rb") as f:
+    svm_model = pickle.load(f)
 
 # Prediction function
-def predict(model_name, sex, pregnant, on_thyroxine, TT4, T3, T4U, FTI, TSH):
+def predict(sex, pregnant, on_thyroxine, TT4, T3, T4U, FTI, TSH):
     try:
-        model = model_map[model_name]
-
+        # Ensure inputs are correctly typed
         sex = int(sex)
         pregnant = int(pregnant)
         on_thyroxine = int(on_thyroxine)
@@ -26,9 +18,9 @@ def predict(model_name, sex, pregnant, on_thyroxine, TT4, T3, T4U, FTI, TSH):
         FTI = float(FTI)
         TSH = float(TSH)
 
-        prediction = model.predict([[sex, pregnant, on_thyroxine, TT4, T3, T4U, FTI, TSH]])
+        prediction = svm_model.predict([[sex, pregnant, on_thyroxine, TT4, T3, T4U, FTI, TSH]])
         label_map = {0: "Hyperthyroid", 1: "Hypothyroid", 2: "Negative"}
-        return f"Prediction using {model_name}: {label_map.get(prediction[0], 'Unknown')}"
+        return f"Prediction: {label_map.get(prediction[0], 'Unknown')}"
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -36,7 +28,6 @@ def predict(model_name, sex, pregnant, on_thyroxine, 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)"),
@@ -47,8 +38,8 @@ demo = gr.Interface(
         gr.Number(label="TSH"),
     ],
     outputs="text",
-    title="Hyperthyroid Prediction (Multi-Model)",
-    description="Select a model and input medical features to predict thyroid condition."
+    title="Hyperthyroid Prediction using SVM",
+    description="Enter 8 medical inputs to predict thyroid condition using an SVM model."
 )
 
 if __name__ == "__main__":