Spaces:
Running
Running
Update app.py
Browse files
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 = [
|
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
|