espejelomar
commited on
Commit
•
f1d0592
1
Parent(s):
633dc2b
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -33,4 +33,4 @@ class ImageClassificationPipeline():
|
|
33 |
{"label": str(self.id2label[i]), "score": float(preds[i])}
|
34 |
for i in range(len(preds))
|
35 |
]
|
36 |
-
return
|
|
|
33 |
{"label": str(self.id2label[i]), "score": float(preds[i])}
|
34 |
for i in range(len(preds))
|
35 |
]
|
36 |
+
return labels
|