nguyennghia0902 commited on
Commit
0b0bdbf
1 Parent(s): b5e36ce

Update QuestionAnswering.py

Browse files
Files changed (1) hide show
  1. QuestionAnswering.py +1 -1
QuestionAnswering.py CHANGED
@@ -3,7 +3,7 @@ import streamlit as st
3
  import tensorflow as tf
4
  from transformers import ElectraTokenizerFast, TFElectraForQuestionAnswering
5
 
6
- model_hf = "nguyennghia0902/electra-small-discriminator_0.0005_32"
7
  tokenizer = ElectraTokenizerFast.from_pretrained(model_hf)
8
  reload_model = TFElectraForQuestionAnswering.from_pretrained(model_hf)
9
 
 
3
  import tensorflow as tf
4
  from transformers import ElectraTokenizerFast, TFElectraForQuestionAnswering
5
 
6
+ model_hf = "nguyennghia0902/electra-small-discriminator_0.0001_16_15e"
7
  tokenizer = ElectraTokenizerFast.from_pretrained(model_hf)
8
  reload_model = TFElectraForQuestionAnswering.from_pretrained(model_hf)
9