fvancesco commited on
Commit
0487285
1 Parent(s): dec748a

remove GPU req.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,7 +17,8 @@ model_name = "fvancesco/tmp_date"
17
  model = AutoModelForMaskedLM.from_pretrained(model_name)
18
  tokenizer = AutoTokenizer.from_pretrained(model_name)
19
  model.eval()
20
- pipe = pipeline('fill-mask', model=model, tokenizer=tokenizer, device=0)
 
21
 
22
  last_mf_dict = None
23
 
 
17
  model = AutoModelForMaskedLM.from_pretrained(model_name)
18
  tokenizer = AutoTokenizer.from_pretrained(model_name)
19
  model.eval()
20
+ #pipe = pipeline('fill-mask', model=model, tokenizer=tokenizer, device=0)
21
+ pipe = pipeline('fill-mask', model=model, tokenizer=tokenizer)
22
 
23
  last_mf_dict = None
24