halictus commited on
Commit
e3f40a0
·
verified ·
1 Parent(s): a92e2de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from PIL import Image
4
  from io import BytesIO
5
  from transformers import pipeline
6
 
7
- model_id = "Honey-Bee-Society/honeybee_bumblebee_vespidae_resnet500"
8
  classifier = pipeline("image-classification", model=model_id)
9
 
10
  def classify_image_from_url(image_url: str):
 
4
  from io import BytesIO
5
  from transformers import pipeline
6
 
7
+ model_id = "Honey-Bee-Society/honeybee_bumblebee_vespidae_resnet50"
8
  classifier = pipeline("image-classification", model=model_id)
9
 
10
  def classify_image_from_url(image_url: str):