jgyasu commited on
Commit
e9c76c6
1 Parent(s): d5fe469

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -430,7 +430,7 @@ fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
430
 
431
  def mask(sentence):
432
  predictions = fill_mask(sentence)
433
- masked_sentences = [f"{predictions[i]['sequence']} \n" for i in range(len(predictions))]
434
  return masked_sentences
435
 
436
  # Function to generate the tree and return the Graphviz source
 
430
 
431
  def mask(sentence):
432
  predictions = fill_mask(sentence)
433
+ masked_sentences = [predictions[i]['sequence'] for i in range(len(predictions))]
434
  return masked_sentences
435
 
436
  # Function to generate the tree and return the Graphviz source