monology commited on
Commit
18ba2a7
·
1 Parent(s): c021659

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -1,13 +1,8 @@
1
  from transformers import pipeline
2
  from PIL import Image
3
 
4
- #device = torch.device("cpu")
5
-
6
- #processor = EfficientNetImageProcessor.from_pretrained("models/dennisjooo/Birds-Classifier-EfficientNetB2")
7
- #model = EfficientNetForImageClassification.from_pretrained("models/dennisjooo/Birds-Classifier-EfficientNetB2").to(device)
8
-
9
  def predict(image):
10
- pipe = pipeline("image-classification", model="models/dennisjooo/Birds-Classifier-EfficientNetB2")
11
  predictions = pipe(img)
12
  return {p["label"]: p["score"] for p in predictions}
13
 
 
1
  from transformers import pipeline
2
  from PIL import Image
3
 
 
 
 
 
 
4
  def predict(image):
5
+ pipe = pipeline("image-classification", model="dennisjooo/Birds-Classifier-EfficientNetB2")
6
  predictions = pipe(img)
7
  return {p["label"]: p["score"] for p in predictions}
8