Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ model = ORTModelForQuestionAnswering.from_pretrained(
|
|
29 |
provider="CPUExecutionProvider"
|
30 |
)
|
31 |
tokenizer = AutoTokenizer.from_pretrained("Xenova/distilbert-base-uncased-distilled-squad")
|
32 |
-
qa_model = pipeline("question-answering", model=model, tokenizer=tokenizer, framework="ort")
|
33 |
|
34 |
def ingest_urls(urls):
|
35 |
"""
|
|
|
29 |
provider="CPUExecutionProvider"
|
30 |
)
|
31 |
tokenizer = AutoTokenizer.from_pretrained("Xenova/distilbert-base-uncased-distilled-squad")
|
32 |
+
qa_model = pipeline("question-answering", model=model, tokenizer=tokenizer, framework="ort", device=0)
|
33 |
|
34 |
def ingest_urls(urls):
|
35 |
"""
|