Update app.py

app.py

@@ -16,7 +16,7 @@ model_names = [
     "0-ma/vit-geometric-shapes-tiny",
 ]
 
-examples = [
+example_images = [
     'example/1_None.jpg',
     'example/2_Circle.jpg',
     'example/3_Triangle.jpg',
@@ -25,7 +25,7 @@ examples = [
     'example/6_Hexagone.jpg'
 ]
 
-labels = [example.split("_")[1].split(".")[0] for example in examples]
+labels = [example.split("_")[1].split(".")[0] for example in example_images]
 
 feature_extractors = {model_name: AutoImageProcessor.from_pretrained(model_name) for model_name in model_names}
 classification_models = {model_name: AutoModelForImageClassification.from_pretrained(model_name) for model_name in model_names}
@@ -49,6 +49,9 @@ def predict(image, selected_model):
 title = "Geometric Shape Classifier"
 description = "Select a model to classify geometric shapes."
 
+# Create examples with both image and default model
+examples = [[img, model_names[0]] for img in example_images]
+
 # Create the Gradio interface
 iface = gr.Interface(
     fn=predict,
@@ -62,4 +65,5 @@ iface = gr.Interface(
     examples=examples
 )
 
-# Launch the interf
+# Launch the interface
+iface.launch()