Update app.py
Browse files
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(
|
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:
|