Update app.py

app.py

@@ -2,8 +2,7 @@ import gradio as gr
 import pickle
 
 # Load models using pickle
-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)
@@ -13,7 +12,6 @@ with open("svm_model.pkl", "rb") as f:
 
 # Map for model selection
 model_map = {
-    "KNN": knn_model,
     "Random Forest": rf_model,
     "SVM": svm_model
 }
@@ -43,7 +41,7 @@ def dis_prediction(model_name, sex, pregnant, TT4, T3, T4U, FTI, TSH):
 demo = gr.Interface(
     fn=dis_prediction,
     inputs=[
-        gr.Dropdown(["SVM", "KNN", "Random Forest"], label="Select Model"),
+        gr.Dropdown(["SVM",  "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.Number(label="TT4"),