Spaces:
Sleeping
Sleeping
datascientist22
commited on
Update app.py
Browse files
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
|
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(
|