Update app.py

app.py

@@ -26,8 +26,12 @@ def titanic(Pclass,Sex,Age,SibSp,Parch,Fare,Embarked):
     input_list.append(Fare)
     input_list.append(Embarked)
     # 'res' is a list of predictions returned as the label.
-    res = model.predict(np.asarray(input_list).reshape(1, -1))     
-    return res
+    res = model.predict(np.asarray(input_list).reshape(1, -1)) 
+    alive_url = "https://illustoon.com/photo/dl/1045.png"
+    died_url = "https://illustoon.com/photo/dl/1052.png"
+    img_path = alive_url if int(res) == 1 else died_url  
+    img = Image.open(requests.get(img_path, stream=True).raw)            
+    return img
         
 demo = gr.Interface(
     fn=titanic,
@@ -43,8 +47,7 @@ demo = gr.Interface(
         gr.inputs.Number(default=1.0, label="Fare [USD]"),
         gr.inputs.Number(default=1.0, label="Embarked [0 (S), 1 (C),  2 (Q)]"),
         ],
-    outputs=gr.Text(value="none")
-    )
+    outputs=gr.Image(type="pil"))
 
 demo.launch()