akdeniz27 commited on
Commit
b074416
1 Parent(s): 9e5d6d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ elif selected_context == "New Context":
57
  def setModel(model_checkpoint):
58
  model = AutoModelForQuestionAnswering.from_pretrained(model_checkpoint)
59
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
60
- return pipeline('"question-answering"', model=model, tokenizer=tokenizer)
61
 
62
  Run_Button = st.button("Run", key=None)
63
  if Run_Button == True:
 
57
  def setModel(model_checkpoint):
58
  model = AutoModelForQuestionAnswering.from_pretrained(model_checkpoint)
59
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
60
+ return pipeline("question-answering", model=model, tokenizer=tokenizer)
61
 
62
  Run_Button = st.button("Run", key=None)
63
  if Run_Button == True: