Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from PIL import Image
 import hopsworks
 
 project = hopsworks.login()
@@ -6,25 +7,25 @@ fs = project.get_feature_store()
 
 dataset_api = project.get_dataset_api()
 
-dataset_api.download("Resources/images/latest_titanic.png", overwrite=True)
-dataset_api.download("Resources/images/actual_titanic.png", overwrite=True)
-dataset_api.download("Resources/images/df_titanic_recent.png", overwrite=True) # naming!
-dataset_api.download("Resources/images/titanic_confusion_matrix.png", overwrite=True) # naming!
+dataset_api.download("Resources/images/latest_iris.png")
+dataset_api.download("Resources/images/actual_iris.png")
+dataset_api.download("Resources/images/df_recent.png")
+dataset_api.download("Resources/images/confusion_matrix.png")
 
 with gr.Blocks() as demo:
     with gr.Row():
-        with gr.Column():
-            gr.Label("Today's Predicted Titanic Survival")
-            input_img = gr.Image("latest_titanic.png", elem_id="predicted-img")
-        with gr.Column():
-            gr.Label("Today's Actual Titanic Survival")
-            input_img = gr.Image("actual_titanic.png", elem_id="actual-img")
+      with gr.Column():
+          gr.Label("Today's Predicted Image")
+          input_img = gr.Image("latest_iris.png", elem_id="predicted-img")
+      with gr.Column():          
+          gr.Label("Today's Actual Image")
+          input_img = gr.Image("actual_iris.png", elem_id="actual-img")        
     with gr.Row():
-        with gr.Column():
-            gr.Label("Recent Prediction History")
-            input_img = gr.Image("df_titanic_recent.png", elem_id="recent-predictions")
-        with gr.Column():
-            gr.Label("Confusion Maxtrix with Historical Prediction Performance")
-            input_img = gr.Image("titanic_confusion_matrix.png", elem_id="confusion-matrix")
+      with gr.Column():
+          gr.Label("Recent Prediction History")
+          input_img = gr.Image("df_recent.png", elem_id="recent-predictions")
+      with gr.Column():          
+          gr.Label("Confusion Maxtrix with Historical Prediction Performance")
+          input_img = gr.Image("confusion_matrix.png", elem_id="confusion-matrix")        
 
-demo.launch()
+demo.launch()