datascientist22 commited on
Commit
45edcab
1 Parent(s): bc9f484

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def extract_text_from_pdfs(files):
45
  return pdf_text
46
 
47
  # Load model and tokenizer
48
- @st.cache_resource(allow_output_mutation=True)
49
  def load_model():
50
  tokenizer = AutoTokenizer.from_pretrained("himmeow/vi-gemma-2b-RAG")
51
  model = AutoModelForCausalLM.from_pretrained(
 
45
  return pdf_text
46
 
47
  # Load model and tokenizer
48
+ @st.cache_resource
49
  def load_model():
50
  tokenizer = AutoTokenizer.from_pretrained("himmeow/vi-gemma-2b-RAG")
51
  model = AutoModelForCausalLM.from_pretrained(