Spaces:
Runtime error
Runtime error
Commit
•
f29fbc7
1
Parent(s):
6d17167
Happy Hugging Face!
Browse files- generator.py +1 -1
generator.py
CHANGED
|
@@ -58,7 +58,7 @@ def run_model(input_string, **generator_args):
|
|
| 58 |
|
| 59 |
tokenizer = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
|
| 60 |
|
| 61 |
-
model =
|
| 62 |
# al_model = pickle.load(open('models/al_model.sav', 'rb'))
|
| 63 |
# al_tokenizer = pickle.load(open('models/al_tokenizer.sav', 'rb'))
|
| 64 |
def QA(question, context):
|
|
|
|
| 58 |
|
| 59 |
tokenizer = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
|
| 60 |
|
| 61 |
+
model = amqa.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
|
| 62 |
# al_model = pickle.load(open('models/al_model.sav', 'rb'))
|
| 63 |
# al_tokenizer = pickle.load(open('models/al_tokenizer.sav', 'rb'))
|
| 64 |
def QA(question, context):
|