Update app.py
Browse files
app.py
CHANGED
|
@@ -102,10 +102,8 @@ def CTXGen(X1, X2, τ, g_num, length_range):
|
|
| 102 |
input_text[mask_position.item()] = predicted_token
|
| 103 |
padded_seq[mask_position.item()] = predicted_token.strip()
|
| 104 |
new_seq = padded_seq
|
| 105 |
-
|
| 106 |
-
print(new_seq)
|
| 107 |
generated_seq = input_text
|
| 108 |
-
print(generated_seq)
|
| 109 |
|
| 110 |
generated_seq[1] = "[MASK]"
|
| 111 |
input_ids = vocab_mlm.__getitem__(generated_seq)
|
|
|
|
| 102 |
input_text[mask_position.item()] = predicted_token
|
| 103 |
padded_seq[mask_position.item()] = predicted_token.strip()
|
| 104 |
new_seq = padded_seq
|
| 105 |
+
|
|
|
|
| 106 |
generated_seq = input_text
|
|
|
|
| 107 |
|
| 108 |
generated_seq[1] = "[MASK]"
|
| 109 |
input_ids = vocab_mlm.__getitem__(generated_seq)
|