Hugging Face's logo

Spaces:
ankz22
/
trash-classifier
Sleeping

App Files Community
ankz22 commited on
Commit
c6200dd
·
1 Parent(s): 6f40eb3

new model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,7 +17,7 @@ augment = transforms.Compose([
17
  # -------------------------------------
18
  # 2. CHARGEMENT DU MODELE FINE-TUNED TRASHNET
19
  # -------------------------------------
20
- MODEL_ID = "mrm8488/resnet50-finetuned-trashnet"
21
  trash_classifier = pipeline(
22
  "image-classification",
23
  model=MODEL_ID,
@@ -25,6 +25,7 @@ trash_classifier = pipeline(
25
  top_k=3
26
  )
27
 
 
28
  # -------------------------------------
29
  # 3. DICTIONNAIRE DE MAPPING
30
  # -------------------------------------
 
17
  # -------------------------------------
18
  # 2. CHARGEMENT DU MODELE FINE-TUNED TRASHNET
19
  # -------------------------------------
20
+ MODEL_ID = "tribber93/my-trash-classification"
21
  trash_classifier = pipeline(
22
  "image-classification",
23
  model=MODEL_ID,
 
25
  top_k=3
26
  )
27
 
28
+
29
  # -------------------------------------
30
  # 3. DICTIONNAIRE DE MAPPING
31
  # -------------------------------------