update

app.py

@@ -7,6 +7,7 @@ def greet(name):
 
 
 name="prophet"
+img="fig2.png"
 project = hopsworks.login()
 
 # Get the dataset API for the current project
@@ -17,7 +18,7 @@ mr = project.get_model_registry()
 #get the version number for the latest model
 version = mr.get_models(name)[-1].version
 # Specify the local file path of the Python script to be uploaded
-img_path = f"Models/{name}/{version}/images/fig1.png"
+img_path = f"Models/{name}/{version}/images/{img}"
 
 # Upload the Python script to the "Models", and overwrite if it already exists
 downloaded_file_path = dataset_api.download(img_path, overwrite=True)
@@ -26,13 +27,13 @@ downloaded_file_path = dataset_api.download(img_path, overwrite=True)
 
 
 def show_image(dummy):
-    image = Image.open("fig1.png")
+    image = Image.open(f"{img}")
     return image
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=show_image,
-    inputs="text",  
+    #inputs="text",  
     outputs="image",
     title="Display PNG Image",
 )