yonadab commited on
Commit
533df08
·
verified ·
1 Parent(s): c82a3d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ app.add_middleware(
18
  )
19
 
20
  # Carga del modelo
21
- processor = AutoImageProcessor.from_pretrained("facebook/dinov2-base")
22
- model = AutoModel.from_pretrained("facebook/dinov2-base")
23
  model.eval()
24
 
25
  # Memoria temporal para almacenar imágenes (podrías usar base de datos si prefieres)
 
18
  )
19
 
20
  # Carga del modelo
21
+ processor = AutoImageProcessor.from_pretrained("facebook/dinov2-small")
22
+ model = AutoModel.from_pretrained("facebook/dinov2-small")
23
  model.eval()
24
 
25
  # Memoria temporal para almacenar imágenes (podrías usar base de datos si prefieres)