Add application file and dependencies

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 from timm import create_model
 import matplotlib.pyplot as plt
 
-class_to_idx = {'Monkeypox': 0, 'Chickenpox': 1, 'Measles': 2, 'Melanoma': 3, 'Herpes': 4}
+class_to_idx = {'Monkeypox': 0, 'Measles': 1, 'Chickenpox': 2, 'Herpes': 3, 'Melanoma': 4}
 idx_to_class = {v: k for k, v in class_to_idx.items()}
 
 # Data transformation
@@ -64,7 +64,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="filepath", label="Upload an image"),
     outputs=[gr.Label(label="Prediction"), gr.Image(label="Prediction Probabilities")],
     title="Skin Lesion Image Classification",
-    description="Upload an image of a skin lesion to get a prediction with confidence percentages. Example images: [Link to Example 1](#), [Link to Example 2](#)",
+    description="Upload an image of a skin lesion to get a prediction . Check out the dataset and paper at: [Link to Example 1](#), [Link to Example 2](#)",
     theme="huggingface",
     live=True
 )