Update app.py

app.py

@@ -24,12 +24,12 @@ def titanic(pclass, sex, age, sibs, par_ch, fare):
     input_list.append(sibs)
     input_list.append(par_ch)
     input_list.append(fare)
-    input_list.append(np.random.choice([0,1], 9))
+    input_list.extend(list(np.random.choice([0,1], 9)))
     # 'res' is a list of predictions returned as the label.
     res = model.predict(np.asarray(input_list).reshape(1, -1)) 
     # We add '[0]' to the result of the transformed 'res', because 'res' is a list, and we only want 
     # the first element.
-    man_url = "https://raw.githubusercontent.com/Tilosmsh/IL2223_lab1/main/images/" + ("survived.jpg" if res[0] else "dead.jpg")
+    man_url = "https://raw.githubusercontent.com/Tilosmsh/IL2223_lab1/main/images/" + ("survived.jpg" if res[0]==1 else "dead.jpg")
     img = Image.open(requests.get(man_url, stream=True).raw)            
     return img