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 |
|