Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ def load_qa_model():
|
|
19 |
|
20 |
tokenizer = AutoTokenizer.from_pretrained('mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es', use_fast="false")
|
21 |
model = TFDistilBertForQuestionAnswering.from_pretrained("mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es", from_pt=True)
|
22 |
-
return model
|
23 |
|
24 |
# 4. Use streamlit to create a web app
|
25 |
def main():
|
@@ -33,7 +33,7 @@ def main():
|
|
33 |
df=cargar_articulos()
|
34 |
articulos_indexados = pd.read_csv('articulos_indexados.csv')
|
35 |
articulos_indexados = articulos_indexados.set_index('Unnamed: 0')
|
36 |
-
qa_model = load_qa_model()
|
37 |
|
38 |
|
39 |
query = st.text_input(
|
|
|
19 |
|
20 |
tokenizer = AutoTokenizer.from_pretrained('mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es', use_fast="false")
|
21 |
model = TFDistilBertForQuestionAnswering.from_pretrained("mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es", from_pt=True)
|
22 |
+
return tokenizer, model
|
23 |
|
24 |
# 4. Use streamlit to create a web app
|
25 |
def main():
|
|
|
33 |
df=cargar_articulos()
|
34 |
articulos_indexados = pd.read_csv('articulos_indexados.csv')
|
35 |
articulos_indexados = articulos_indexados.set_index('Unnamed: 0')
|
36 |
+
tokenizer, qa_model = load_qa_model()
|
37 |
|
38 |
|
39 |
query = st.text_input(
|