titipata commited on
Commit
bdfc761
1 Parent(s): 19149e5

Minor update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def inference(abstract: str):
24
  for (idx, out) in enumerate(output):
25
  pred = out.argmax().item()
26
  if pred:
27
- claims.append(lines[idx])
28
  if len(claims) > 0:
29
  return ".\n".join(claims)
30
  return "No claims were made here"
 
24
  for (idx, out) in enumerate(output):
25
  pred = out.argmax().item()
26
  if pred:
27
+ claims.append(sents[idx])
28
  if len(claims) > 0:
29
  return ".\n".join(claims)
30
  return "No claims were made here"