Lisibonny commited on
Commit
2d4b1b0
1 Parent(s): 692954f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,8 +45,8 @@ def crear_indice():
45
  def load_qa_model():
46
 
47
  tokenizer = AutoTokenizer.from_pretrained('dccuchile/bert-base-spanish-wwm-cased', use_fast="false")
48
- model = AutoModelForQuestionAnswering.from_pretrained("Lisibonny/modelo_qa_beto_squad")
49
- #model = TFAutoModel.from_pretrained("Lisibonny/modelo_qa_beto_squad")
50
  return tokenizer, model
51
 
52
  # 4. Use streamlit to create a web app
 
45
  def load_qa_model():
46
 
47
  tokenizer = AutoTokenizer.from_pretrained('dccuchile/bert-base-spanish-wwm-cased', use_fast="false")
48
+ #model = AutoModelForQuestionAnswering.from_pretrained("Lisibonny/modelo_qa_beto_squad")
49
+ model = TFAutoModelForQuestionAnswering.from_pretrained("Lisibonny/modelo_qa_beto_squad")
50
  return tokenizer, model
51
 
52
  # 4. Use streamlit to create a web app