Spaces:
Sleeping
Sleeping
Update App.py
Browse files
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 =
|
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 |
|