updates

app.py

@@ -17,23 +17,23 @@ 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/{img}"
 
-# Upload the Python script to the "Models", and overwrite if it already exists
-downloaded_file_path = dataset_api.download(img_path, overwrite=True)
 
 
 
+def show_image(img):
+    # Specify the local file path of the Python script to be uploaded
+    img_path = f"Models/{name}/{version}/images/{img}.png"
 
-def show_image(dummy):
-    image = Image.open(f"{img}")
+    # Upload the Python script to the "Models", and overwrite if it already exists
+    downloaded_file_path = dataset_api.download(img_path, overwrite=True)
+    image = Image.open(f"{img}.png")
     return image
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=show_image,
-    inputs="text",  
+    inputs=gr.Textbox(default="fig2", label="Image Name"),  
     outputs="image",
     title="Display PNG Image",
 )