Lisibonny commited on
Commit
3a6124d
1 Parent(s): a49cecf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(