yangheng commited on
Commit
f676dc4
1 Parent(s): 009d64e

Update app.py

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