yohannesdesta commited on
Commit
5948f46
·
1 Parent(s): 08a73ed
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -19,12 +19,18 @@ def normalize(value, cast=None):
19
 
20
 
21
  def predict_job(job_data):
22
- df = pd.DataFrame([job_data])
 
 
 
 
 
 
 
23
 
24
  # Predict
25
- probability = model.predict_proba(df)[0][1] # Probability of HIRED
26
  prediction = "HIRED" if probability > 0.5 else "NOT_HIRED"
27
-
28
  return {
29
  'prediction': prediction,
30
  'hire_probability': f"{probability*100:.1f}",
 
19
 
20
 
21
  def predict_job(job_data):
22
+ import numpy as np
23
+
24
+ # Replace None with np.nan to ensure numeric dtype
25
+ clean_data = {k: (np.nan if v is None else v) for k, v in job_data.items()}
26
+ df = pd.DataFrame([clean_data])
27
+
28
+ # Force all columns to float64 to match training data
29
+ df = df.astype('float64')
30
 
31
  # Predict
32
+ probability = model.predict_proba(df)[0][1]
33
  prediction = "HIRED" if probability > 0.5 else "NOT_HIRED"
 
34
  return {
35
  'prediction': prediction,
36
  'hire_probability': f"{probability*100:.1f}",