Spaces:
Running
Running
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 |
|