papluca commited on
Commit
9f1dbb5
·
verified ·
1 Parent(s): 58128e1

[Fix] Typo in variable name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 pred}
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