kajsndaui

app.py

@@ -1,39 +1,16 @@
-from typing import Any
 import gradio as gr
 import hopsworks
 from PIL import Image
 
 project = hopsworks.login()
-#fs = project.get_feature_store()
-
-dataset_api = project.get_dataset_api()
-
-dataset_api.download("Resources/images/placeholder.png", overwrite=True)
-dataset_api.download("Resources/images/placeholder.png", overwrite=True)
-dataset_api.download("Resources/images/placeholder.png", overwrite=True)
-dataset_api.download("Resources/images/placeholder.png", overwrite=True)
-
-""" def load_images():
-    project = hopsworks.login()
-    dataset_api = project.get_dataset_api()
-
-    dataset_api.download("Resources/images/latest_wine.png", overwrite=True)
-    dataset_api.download("Resources/images/actual_wine.png", overwrite=True)
-    dataset_api.download("Resources/images/df_wine_recent.png", overwrite=True)
-    dataset_api.download("Resources/images/wine_confusion_matrix.png", overwrite=True)
-
-    return Image.open("wine_confusion_matrix.png") """
-
 
 class ImageLoad:
     def __init__(self, path, project) -> None:
         self.path = path
         self.image_name = path[path.rfind('/') + 1:]
         self.project = project
-        print(self.image_name)
 
     def __call__(self):
-        #project = hopsworks.login()
         dataset_api = self.project.get_dataset_api()
         dataset_api.download(self.path, overwrite=True)
         return Image.open(self.image_name)
@@ -53,13 +30,7 @@ with gr.Blocks() as demo:
           gr.Label("Recent Prediction History")
           input_img = gr.Image(value=ImageLoad("Resources/images/df_wine_recent.png", project), elem_id="recent-predictions")
       with gr.Column():          
-          gr.Label("Confusion Maxtrix with Historical Prediction Performance")
-          #input_img = gr.Image("wine_confusion_matrix.png", elem_id="confusion-matrix")     
-          """ image = gr.Image(show_label=False)
-          demo.load(fn=load_images, inputs=None, outputs=image,
-                show_progress=False)  """  
+          gr.Label("Confusion Maxtrix with Historical Prediction Performance")  
           input_img = gr.Image(value=ImageLoad("Resources/images/wine_confusion_matrix.png", project), elem_id="recent-predictions")
-    #demo.load(load_images)
 
-#load_images()
 demo.launch(share = True)