Shriharsh commited on
Commit
bc9fd78
·
verified ·
1 Parent(s): 7215005

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
  """