Docty commited on
Commit
0747d3a
·
verified ·
1 Parent(s): 40a16da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from PIL import Image
3
  from transformers import pipeline
4
 
5
 
6
- #classifier = pipeline("image-classification", model="/content/mango-samu")
7
- classifier = pipeline("image-classification", model="Docty/mango-samu")
8
 
9
  def classify_image(img):
10
  if not isinstance(img, Image.Image):
 
3
  from transformers import pipeline
4
 
5
 
6
+
7
+ classifier = pipeline("image-classification", model="Docty/mangoes")
8
 
9
  def classify_image(img):
10
  if not isinstance(img, Image.Image):