Mary12 commited on
Commit
e815c6a
·
1 Parent(s): 03c0c73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -50,11 +50,7 @@ def extract_text_from_doc(file_path):
50
  def model(model_name):
51
  tokenizer = AutoTokenizer.from_pretrained(model_name)
52
  model = AutoModelForQuestionAnswering.from_pretrained(model_name,return_dict = False)
53
- model_pipeline = pipeline4
54
- "question-answering",
55
- model = model,
56
- tokenizer = tokenizer
57
- )
58
 
59
  return model_pipeline
60
 
 
50
  def model(model_name):
51
  tokenizer = AutoTokenizer.from_pretrained(model_name)
52
  model = AutoModelForQuestionAnswering.from_pretrained(model_name,return_dict = False)
53
+ model_pipeline = pipeline("question-answering",model = model,tokenizer = tokenizer)
 
 
 
 
54
 
55
  return model_pipeline
56