Hugging Face's logo

Spaces:
vinayakdev
/
qa-generator
Runtime error

App Files Community
vinayakdev commited on
Commit
ae4c218
·
1 Parent(s): 13b79aa

Remove cache at one place

Browse files
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -39,7 +39,7 @@ def load_model():
39
  hftokenizer = T5TokenizerFast.from_pretrained("t5-base")
40
  tokenizer = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
41
  model = amqa.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
42
- @st.cache(suppress_st_warning=True)
43
  load_model()
44
 
45
  def run_model(input_string, **generator_args):
 
39
  hftokenizer = T5TokenizerFast.from_pretrained("t5-base")
40
  tokenizer = att.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
41
  model = amqa.from_pretrained("ahotrod/albert_xxlargev1_squad2_512")
42
+
43
  load_model()
44
 
45
  def run_model(input_string, **generator_args):