papluca commited on
Commit
58128e1
·
verified ·
1 Parent(s): 7509d09

Update to new version of transformers library

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