MODLI commited on
Commit
d32e8d3
·
verified ·
1 Parent(s): 9d0e788

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ seg_pipe = pipeline("image-segmentation", model="mattmdjaga/segformer_b2_clothes
13
 
14
  print("Loading fashion classification model...")
15
  # Modèle de classification spécialisé dans la mode (FashionCLIP)
16
- class_pipe = pipeline("zero-shot-image-classification", model="edwardjross/fashion-clip")
 
17
  print("Models loaded successfully!")
18
 
19
  # --- Fonctions de Prétraitement ---
 
13
 
14
  print("Loading fashion classification model...")
15
  # Modèle de classification spécialisé dans la mode (FashionCLIP)
16
+ # Un modèle qui fait de la classification d'images de mode
17
+ class_pipe = pipeline("image-classification", model="asafaya/convnext-tiny-224-fashion")
18
  print("Models loaded successfully!")
19
 
20
  # --- Fonctions de Prétraitement ---