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 |
|