Update app.py

app.py

@@ -32,9 +32,14 @@ def titanic(PassengerId, Pclass, Numeric_sex, Age, Fare):
     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.
-        
+    l = int(res)   
+    if(l == 0):
+            img = Image.open("deadImage.png")            
+    else:
+            img = Image.open("survivedImage.png")
     #flower_url = "https://raw.githubusercontent.com/featurestoreorg/serverless-ml-course/main/src/01-module/assets/" + res[0] + ".png"
-    img = Image.open("deadImage.png")#requests.get(flower_url, stream=True).raw)            
+   # img = Image.open("deadImage.png")#requests.get(flower_url, stream=True).raw)            
+    
     return img
         
 demo = gr.Interface(