MasoudSamaei commited on
Commit
1fa6791
1 Parent(s): 0e8204c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -9,22 +9,25 @@ def make_prediction(al, type, dr, sn, percent, tmax):
9
  clf = pickle.load(f)
10
  preds = clf.predict([[type, dr, sn, percent]])
11
  return f"""Shear strength after improvement: {round((preds.squeeze() * tmax),2)}
12
- Improvement percentage is {round((preds.squeeze()-1)*100,2)}%"""
13
  elif al == "Random Forest" :
14
  with open("random_forest.pkl", "rb") as f:
15
  clf = pickle.load(f)
16
  preds = clf.predict([[type, dr, sn, percent]])
17
- return f"Improvement percentage would be around {round((preds.squeeze()-1)*100, 2)}% and shear strength improvement predicted by Random Forest is {round((preds.squeeze() * tmax), 2)}"
 
18
  elif al == "XGBoost" :
19
  with open("XGBoost.pkl", "rb") as f:
20
  clf = pickle.load(f)
21
  preds = clf.predict([[type, dr, sn, percent]])
22
- return f"Improvement percentage would be around {round((preds.squeeze()-1)*100, 2)}% and shear strength improvement predicted by XGBoost is {round((preds.squeeze() * tmax), 2)}"
 
23
  elif al == "AdaBoost" :
24
  with open("AdaBoost.pkl", "rb") as f:
25
  clf = pickle.load(f)
26
  preds = clf.predict([[type, dr, sn, percent]])
27
- return f"Improvement percentage would be around {round((preds.squeeze()-1)*100, 2)}% and shear strength improvement predicted by AdaBoost is {round((round(preds.squeeze() * tmax), 2))}"
 
28
 
29
 
30
  import gradio as gr
 
9
  clf = pickle.load(f)
10
  preds = clf.predict([[type, dr, sn, percent]])
11
  return f"""Shear strength after improvement: {round((preds.squeeze() * tmax),2)}
12
+ Improvement percentage is {round((preds.squeeze()-1)*100,2)}%"""
13
  elif al == "Random Forest" :
14
  with open("random_forest.pkl", "rb") as f:
15
  clf = pickle.load(f)
16
  preds = clf.predict([[type, dr, sn, percent]])
17
+ return f"""Shear strength after improvement: {round((preds.squeeze() * tmax),2)}
18
+ Improvement percentage is {round((preds.squeeze()-1)*100,2)}%"""
19
  elif al == "XGBoost" :
20
  with open("XGBoost.pkl", "rb") as f:
21
  clf = pickle.load(f)
22
  preds = clf.predict([[type, dr, sn, percent]])
23
+ return f"""Shear strength after improvement: {round((preds.squeeze() * tmax),2)}
24
+ Improvement percentage is {round((preds.squeeze()-1)*100,2)}%"""
25
  elif al == "AdaBoost" :
26
  with open("AdaBoost.pkl", "rb") as f:
27
  clf = pickle.load(f)
28
  preds = clf.predict([[type, dr, sn, percent]])
29
+ return f"""Shear strength after improvement: {round((preds.squeeze() * tmax),2)}
30
+ Improvement percentage is {round((preds.squeeze()-1)*100,2)}%"""
31
 
32
 
33
  import gradio as gr