Spaces:

ariG23498
/

food-classification

Running

ariG23498 HF Staff commited on Jan 13

Commit

b455d78

verified ·

1 Parent(s): 37db87d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import pipeline
 pipe = pipeline(
     "image-classification",
-    model="ariG23498/vit_base_patch16_224.orig_in21k_ft_in1k.ft_food101"
 )
 def classify(image):

 pipe = pipeline(
     "image-classification",
+    model="ariG23498/vit_base_patch16_224.augreg2_in21k_ft_in1k.ft_food101"
 )
 def classify(image):