Spaces:
Runtime error
Runtime error
Commit
·
8336e59
1
Parent(s):
bcad6bc
Update generator.py
Browse files- generator.py +2 -2
generator.py
CHANGED
@@ -39,9 +39,9 @@ def load_model():
|
|
39 |
model = pickle.load(open('model.sav','rb'))
|
40 |
tok = AlbertTokenizer.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
|
41 |
# return hfm, hft,tok, model
|
42 |
-
return hfm, hft,
|
43 |
|
44 |
-
hfmodel, hftokenizer,
|
45 |
|
46 |
def run_model(input_string, **generator_args):
|
47 |
generator_args = {
|
|
|
39 |
model = pickle.load(open('model.sav','rb'))
|
40 |
tok = AlbertTokenizer.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
|
41 |
# return hfm, hft,tok, model
|
42 |
+
return hfm, hft, tok, model
|
43 |
|
44 |
+
hfmodel, hftokenizer, tokenizer, model = load_model()
|
45 |
|
46 |
def run_model(input_string, **generator_args):
|
47 |
generator_args = {
|