Spaces:
Sleeping
Sleeping
fvancesco
commited on
Commit
·
0487285
1
Parent(s):
dec748a
remove GPU req.
Browse files
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 |
|