fix

app.py

@@ -49,13 +49,13 @@ if __name__ == '__main__':
         inputs=[
             gr.Image(type="filepath", label="Image"),
             gr.Slider(0.0, 1.0, step=0.05, value=0.4, label="Threshold"),
-            gr.Dropdown(choices=['dino_vits', 'dino_vitb', 'dinov2_vits', 'dinov2_vitb'], value='dino_vitb', label="SSL Features"),
+            gr.Dropdown(choices=['dino_vits', 'dino_vitb', 'dinov2_vits', 'dinov2_vitb'], value='dino_vits', label="SSL Features"),
         ],
         outputs=gr.Image(label="PriMaPs"),
         title="PriMaPs Demo",
         description="Upload an image and adjust the threshold to visualize PriMaPs.",
         examples=[
-            ["example.jpg", 0.4, 'dino_vitb'],
+            ["example.jpg", 0.4, 'dino_vits'],
         ]
     )