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
|