shams1992 commited on
Commit
97e366e
·
1 Parent(s): 19f3906

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ nlp = pipeline('question-answering',
6
  tokenizer='databricks/dolly-v2-3b',
7
  torch_dtype=torch.bfloat16,
8
  trust_remote_code=True,
9
- device_map="auto"))
10
 
11
  def qnamodel(context,question):
12
  question_set = {'context':context,'question':question}
 
6
  tokenizer='databricks/dolly-v2-3b',
7
  torch_dtype=torch.bfloat16,
8
  trust_remote_code=True,
9
+ device_map="auto")
10
 
11
  def qnamodel(context,question):
12
  question_set = {'context':context,'question':question}