Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,11 +7,11 @@ fill_mask = pipeline("fill-mask", model=model_ckpt)
|
|
7 |
|
8 |
|
9 |
def cloze_task(text):
|
10 |
-
preds = fill_mask(text)
|
11 |
-
return preds
|
12 |
|
13 |
-
|
14 |
-
|
15 |
description="Let's see if BERT has learnt how to write Keras!"
|
16 |
title="Keras BERT v1",
|
17 |
interface = gr.Interface(fn= cloze_task,
|
|
|
7 |
|
8 |
|
9 |
def cloze_task(text):
|
10 |
+
preds = fill_mask(text)
|
11 |
+
return preds[0]["sequence"]
|
12 |
|
13 |
+
[{'sequence': 'from tensorflow.keras import layers', 'score': 0.9447882771492004, 'token': 455, 'token_str': ' layers'}, {'sequence': 'from tensorflow.keras import optimizers', 'score': 0.010178953409194946, 'token': 9110, 'token_str': ' optimizers'}, {'sequence': 'from tensorflow.keras import regularizers', 'score': 0.008282472379505634, 'token': 14453, 'token_str': ' regularizers'}, {'sequence': 'from tensorflow.keras import losses', 'score': 0.004894345533102751, 'token': 3114, 'token_str': ' losses'}, {'sequence': 'from tensorflow.keras import Model', 'score': 0.003724579466506839, 'token': 2663, 'token_str': ' Model'}]
|
14 |
+
|
15 |
description="Let's see if BERT has learnt how to write Keras!"
|
16 |
title="Keras BERT v1",
|
17 |
interface = gr.Interface(fn= cloze_task,
|