ASNVS commited on
Commit
eef8fb4
·
verified ·
1 Parent(s): 13b4574

updated ones

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -38,6 +38,9 @@ joblib.dump(label_encoders, "label_encoders.pkl")
38
 
39
  # Prediction function
40
  def predict_colleges(category, gender, rank, region):
 
 
 
41
  # Load label encoders
42
  label_encoders = joblib.load("label_encoders.pkl")
43
 
@@ -74,7 +77,7 @@ demo = gr.Interface(
74
  inputs=[
75
  gr.Dropdown(choices=["OC", "BC", "SC", "ST"], label="Category"),
76
  gr.Radio(choices=["Male", "Female"], label="Gender"),
77
- gr.Number(label="Rank"),
78
  gr.Dropdown(choices=["AU", "SV"], label="Region")
79
  ],
80
  outputs=gr.Dataframe(headers=["College Name", "Branch"]),
 
38
 
39
  # Prediction function
40
  def predict_colleges(category, gender, rank, region):
41
+ if not isinstance(rank, (int, float)) or rank < 0:
42
+ return "Invalid Rank: Please enter a valid positive number."
43
+
44
  # Load label encoders
45
  label_encoders = joblib.load("label_encoders.pkl")
46
 
 
77
  inputs=[
78
  gr.Dropdown(choices=["OC", "BC", "SC", "ST"], label="Category"),
79
  gr.Radio(choices=["Male", "Female"], label="Gender"),
80
+ gr.Number(label="Rank", precision=0), # Restrict to whole numbers only
81
  gr.Dropdown(choices=["AU", "SV"], label="Region")
82
  ],
83
  outputs=gr.Dataframe(headers=["College Name", "Branch"]),