Update app.py

app.py

@@ -52,15 +52,10 @@ def incident(pclass, sex, age, sibsp, parch, fare, embarked):
     input_list.append(fare)
     input_list.append(pclass)
 
-    # '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.
-    if res[0] == 1.0:
-        survive_url = "https://raw.githubusercontent.com/Hope-Liang/ID2223Lab1/main/serverless-ml-titanic/images/survived.png"
-    else:
-        survive_url = "https://raw.githubusercontent.com/Hope-Liang/ID2223Lab1/main/serverless-ml-titanic/images/died.png"
-    img = Image.open(requests.get(survive_url, stream=True).raw)
+    incident = model.predict(np.asarray(input_list).reshape(1, -1))
+    incident_url = "https://raw.githubusercontent.com/Hope-Liang/ID2223Project/main/images/" + incident[0] + ".png"
+    img = Image.open(requests.get(incident_url, stream=True).raw)
+    
     return img