Update app.py

app.py

@@ -62,12 +62,7 @@ def set_example_url(example: list) -> dict:
 
 title = """<h1 id="title">Plant Health Classification with ViT</h1>"""
 
-gr.Image(pathlib.Path('images/Healthy.png'),label = 'Healthy Plant') 
-
-gr.Image(pathlib.Path('images/sickie.png'),label = 'Infected Plant')
-
-from IPython.display import display, Image
-display(Image(filename=pathlib.Path('images/sickie.png')))
+gr.Markdown('''![alt text](images/sickie.png "Title")''')
 
 description = """
 This Plant Health classifier app was built to detect the health of plants using images of leaves by fine-tuning a Vision Transformer (ViT) [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the [Beans](https://huggingface.co/datasets/beans) dataset.