Hannnnnah commited on
Commit
b8799f9
1 Parent(s): 04472f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def titanic(Pclass,Sex,Age,SibSp,Parch,Fare,Embarked):
29
  res = model.predict(np.asarray(input_list).reshape(1, -1))
30
  alive_url = "https://illustoon.com/photo/dl/1045.png"
31
  died_url = "https://illustoon.com/photo/dl/1052.png"
32
- img_path = alive_url if int(res) == 1 else died_url
33
  img = Image.open(requests.get(img_path, stream=True).raw)
34
  return img
35
 
 
29
  res = model.predict(np.asarray(input_list).reshape(1, -1))
30
  alive_url = "https://illustoon.com/photo/dl/1045.png"
31
  died_url = "https://illustoon.com/photo/dl/1052.png"
32
+ img_path = alive_url if int(res[0]) == 1 else died_url
33
  img = Image.open(requests.get(img_path, stream=True).raw)
34
  return img
35