fxcb commited on
Commit
e2fc846
1 Parent(s): b200481

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 a esquiar a <mask>."
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