MasoudSamaei commited on
Commit
90fb362
1 Parent(s): 1fa6791

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -9,25 +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"""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
 
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: {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: {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: {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: {round((preds.squeeze()-1)*100,2)}%"""
31
 
32
 
33
  import gradio as gr