decodingchris commited on
Commit
2ca9b3b
Β·
verified Β·
1 Parent(s): 4a2fe6b

Change qa model to distilbert cased

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import spaces
3
 
4
  from transformers import pipeline
5
 
6
- pipe = pipeline(task="question-answering", model="decodingchris/distilbert-base-uncased-finetuned-squad-v2")
7
 
8
  @spaces.GPU
9
  def get_answer(context, question):
 
3
 
4
  from transformers import pipeline
5
 
6
+ pipe = pipeline(task="question-answering", model="distilbert/distilbert-base-cased-distilled-squad")
7
 
8
  @spaces.GPU
9
  def get_answer(context, question):