variablesynthetic

app.py

@@ -2,7 +2,22 @@ import gradio as gr
 from PIL import Image
 import hopsworks
 
-# teste
+
+SYNTHETIC = False
+
+latestSurvivorImage = 'latest_survivor_pred.png'
+latestSurvivorPred = 'latest_survivor_label_pred.png'
+latestSurvivorLabel = 'latest_survivor_label_actual.png'
+recentHistory = 'df_recent_titanic.png'
+confusionMatrix = 'confusion_matrix.png'
+
+if SYNTHETIC:
+    latestSurvivorImage += '_synthetic'
+    latestSurvivorPred += '_synthetic'
+    latestSurvivorLabel += '_synthetic'
+    recentHistory += '_synthetic'
+    confusionMatrix += '_synthetic'
+
 with gr.Blocks() as demo:
     project = hopsworks.login()
     fs = project.get_feature_store()
@@ -10,39 +25,39 @@ with gr.Blocks() as demo:
     dataset_api = project.get_dataset_api()
 
     print('Downloading...')
-    dataset_api.download("Resources/images/latest_survivor_pred_synthetic.png")
+    dataset_api.download(f"Resources/images/{latestSurvivorImage}")
     dataset_api.download(
-        "Resources/images/latest_survivor_label_pred_synthetic.png")
+        f"Resources/images/{latestSurvivorPred}")
     dataset_api.download(
-        "Resources/images/latest_survivor_label_actual_synthetic.png")
-    dataset_api.download("Resources/images/df_recent_titanic_synthetic.png")
-    dataset_api.download("Resources/images/confusion_matrix_synthetic.png")
+        f"Resources/images/{latestSurvivorLabel}")
+    dataset_api.download(f"Resources/images/{recentHistory}")
+    dataset_api.download(f"Resources/images/{confusionMatrix}")
 
     with gr.Column():
         gr.Label("Today's passenger")
-        input_img = gr.Image("latest_survivor_pred_synthetic.png",
+        input_img = gr.Image(f"{latestSurvivorImage}",
                              elem_id="passenger-img").style(
             height='100', rounded=False)
         with gr.Row():
             with gr.Column():
                 gr.Label("Today's predicted survival")
                 input_img = gr.Image(
-                    "latest_survivor_label_pred_synthetic.png", elem_id="predicted-img").style(
+                    f"{latestSurvivorPred}", elem_id="predicted-img").style(
                     height='100', rounded=False)
             with gr.Column():
                 gr.Label("Today's actual survival")
                 input_img = gr.Image(
-                    "latest_survivor_label_actual_synthetic.png", elem_id="actual-img").style(
+                    f"{latestSurvivorLabel}", elem_id="actual-img").style(
                     height='100', rounded=False)
         with gr.Row():
             with gr.Column():
                 gr.Label("Recent Prediction History")
                 input_img = gr.Image(
-                    "df_recent_titanic_synthetic.png", elem_id="recent-predictions")
+                    f"{recentHistory}", elem_id="recent-predictions")
             with gr.Column():
                 gr.Label(
                     "Confusion Maxtrix with Historical Prediction Performance")
-                input_img = gr.Image("confusion_matrix_synthetic.png",
+                input_img = gr.Image(f"{confusionMatrix}",
                                      elem_id="confusion-matrix")
 
 demo.launch()