Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ tokenizer_hf = AutoTokenizer.from_pretrained('projecte-aina/roberta-base-ca-v2')
|
|
| 7 |
model = AutoModelForMaskedLM.from_pretrained('projecte-aina/roberta-base-ca-v2')
|
| 8 |
model.eval()
|
| 9 |
pipeline = FillMaskPipeline(model, tokenizer_hf)
|
| 10 |
-
text = f"Tinc pressa, marxo
|
| 11 |
res_hf = pipeline(text)
|
| 12 |
res = ([r['token_str'] for r in res_hf])
|
| 13 |
|
|
|
|
| 7 |
model = AutoModelForMaskedLM.from_pretrained('projecte-aina/roberta-base-ca-v2')
|
| 8 |
model.eval()
|
| 9 |
pipeline = FillMaskPipeline(model, tokenizer_hf)
|
| 10 |
+
text = f"Tinc pressa, marxo al MACBA a veure l'exposició <mask>."
|
| 11 |
res_hf = pipeline(text)
|
| 12 |
res = ([r['token_str'] for r in res_hf])
|
| 13 |
|