Update app.py

app.py

@@ -81,9 +81,10 @@ def titanic(pclass, sex, age, fare, embarked, familysize, appellation, cabin):
 
     # 'res' is a list of predictions returned as the label.
     res = model.predict(np.asarray(input_list).reshape(1, -1))
+    res = res.astype(int)
     # We add '[0]' to the result of the transformed 'res', because 'res' is a list, and we only want
     # the first element.
-    titanic_url = "https://github.com/Qinglin2000/ID2223" + str(res[0]) + ".png?raw=true"
+    titanic_url = "https://github.com/Qinglin2000/ID2223/" + str(res[0]) + ".png"
     img = Image.open(requests.get(titanic_url, stream=True).raw)
     return img