Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -162,7 +162,7 @@ def perform_aste_inference(text, dataset):
|
|
| 162 |
result = triplet_extractor.predict(text)
|
| 163 |
|
| 164 |
pred_triplets = pd.DataFrame(result["Triplets"])
|
| 165 |
-
true_triplets = pd.DataFrame(result["True Triplets"])
|
| 166 |
return pred_triplets, true_triplets, "{}".format(text.split("####")[0])
|
| 167 |
|
| 168 |
|
|
|
|
| 162 |
result = triplet_extractor.predict(text)
|
| 163 |
|
| 164 |
pred_triplets = pd.DataFrame(result["Triplets"])
|
| 165 |
+
true_triplets = pd.DataFrame(result["True Triplets"]) if result["True Triplets"] else None
|
| 166 |
return pred_triplets, true_triplets, "{}".format(text.split("####")[0])
|
| 167 |
|
| 168 |
|