Honey Bee Society commited on
Commit
e320744
·
1 Parent(s): a31153b
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +5 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from PIL import Image
4
  from io import BytesIO
5
  from transformers import pipeline
6
 
 
7
  # 1. Load a pretrained ResNet-50 from the Hugging Face Hub
8
  model_id = "halictus/resnet50_honeybee"
9
  classifier = pipeline("image-classification", model=model_id)
 
4
  from io import BytesIO
5
  from transformers import pipeline
6
 
7
+ #new
8
  # 1. Load a pretrained ResNet-50 from the Hugging Face Hub
9
  model_id = "halictus/resnet50_honeybee"
10
  classifier = pipeline("image-classification", model=model_id)
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ gradio
2
+ transformers
3
+ torch
4
+ Pillow
5
+ requests