Spaces:
Sleeping
Sleeping
Update to new version of transformers library
Browse files
app.py
CHANGED
|
@@ -13,7 +13,6 @@ def predict(text: str) -> dict:
|
|
| 13 |
"""Compute predictions for text."""
|
| 14 |
preds = pipe(text, top_k=3, truncation=True, max_length=128)
|
| 15 |
if preds:
|
| 16 |
-
pred = preds[0]
|
| 17 |
return {p["label"]: float(p["score"]) for p in pred}
|
| 18 |
else:
|
| 19 |
return None
|
|
|
|
| 13 |
"""Compute predictions for text."""
|
| 14 |
preds = pipe(text, top_k=3, truncation=True, max_length=128)
|
| 15 |
if preds:
|
|
|
|
| 16 |
return {p["label"]: float(p["score"]) for p in pred}
|
| 17 |
else:
|
| 18 |
return None
|