Update app.py

app.py

@@ -17,6 +17,16 @@ model_names = [
     "0-ma/vit-geometric-shapes-tiny",
 ]
 
+examples = [
+    'example/1_None.jpg',
+    'example/2_Circle.jpg',
+    'example/3_Triangle.jpg',
+    'example/4_Square.jpg',
+    'example/5_Pentagone.jpg',
+    'example/6_Hexagone.jpg'
+]
+
+labels = [example.split("_")[1].spilt(".")[0] for example in examples]
 # Load the default model
 #feature_extractor = AutoImageProcessor.from_pretrained(models["Tiny Model"])
 #model = AutoModelForImageClassification.from_pretrained(models["Tiny Model"])
@@ -51,14 +61,6 @@ def predict(image):
 title = "Geometric Shape Classifier"
 description = "Select a model to classify geometric shapes."
 
-examples = [
-    'example/1_None.jpg',
-    'example/2_Circle.jpg',
-    'example/3_Triangle.jpg',
-    'example/4_Square.jpg',
-    'example/5_Pentagone.jpg',
-    'example/6_Hexagone.jpg'
-]
 
 # Adding a dropdown for model selection
 gr.Interface(