Zakia commited on
Commit
ad7296c
1 Parent(s): 873b4ad
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -19,10 +19,14 @@ def predict(model, input_df):
19
  amputation_risk = ""
20
  if predict_label == 1:
21
  amputation_risk = "YES"
22
- output = "<div style='background-color:green;'>"+"AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)+"</div>"
 
 
23
  else:
24
  amputation_risk = "NO"
25
- output = "<div style='background-color:green;'>"+"AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)+"</div>"
 
 
26
 
27
 
28
  return output#"AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)
 
19
  amputation_risk = ""
20
  if predict_label == 1:
21
  amputation_risk = "YES"
22
+ output = "AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)
23
+ html = displacy.render(output, style="dep", page=True)
24
+ html = "<div style='background-color:red;'>" + html + "</div>"
25
  else:
26
  amputation_risk = "NO"
27
+ output = "AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)
28
+ html = displacy.render(output, style="dep", page=True)
29
+ html = "<div style='background-color:green;'>" + html + "</div>"
30
 
31
 
32
  return output#"AMPUTATION RISK: " + amputation_risk + " SCORE: "+str(predict_score)