Minor update
Browse files
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(
|
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"
|