Spaces:
Sleeping
Sleeping
[Fix] Typo in variable name
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ 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 |
-
return {p["label"]: float(p["score"]) for p in
|
| 17 |
else:
|
| 18 |
return None
|
| 19 |
|
|
|
|
| 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 preds}
|
| 17 |
else:
|
| 18 |
return None
|
| 19 |
|