Asseh commited on
Commit
1952c57
1 Parent(s): 6d138f7

Update App.py

Browse files
Files changed (1) hide show
  1. App.py +1 -1
App.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  classifier = pipeline("image-classification", model="Asseh/Ball_Classification")
5
 
6
  def predict(input_img):
7
- predictions = classifier(input_img)
8
  return input_img, {p["label"]: p["score"] for p in predictions}
9
 
10
 
 
4
  classifier = pipeline("image-classification", model="Asseh/Ball_Classification")
5
 
6
  def predict(input_img):
7
+ predictions = pipe(input_img)
8
  return input_img, {p["label"]: p["score"] for p in predictions}
9
 
10