Hugging Face's logo

Spaces:
tasmimulhuda
/
fastapi-rag-qa
Runtime error

App Files Community
tasmimulhuda commited on
Commit
4609c12
·
1 Parent(s): fbf063e

chche dir

Browse files
Files changed (1) hide show
  1. app/rag_pipeline/model_initializer.py +1 -0
app/rag_pipeline/model_initializer.py CHANGED
@@ -36,6 +36,7 @@ def initialize_models(openai_api_key=None,model_id=None, model_basename=None):
36
  llm_model = HuggingFacePipeline.from_model_id(
37
  model_id= "gpt2", #"google/flan-t5-small",
38
  task="text-generation",
 
39
  )
40
 
41
  #TheBloke/Mistral-7B-v0.1-GGUF
 
36
  llm_model = HuggingFacePipeline.from_model_id(
37
  model_id= "gpt2", #"google/flan-t5-small",
38
  task="text-generation",
39
+ cache_dir=CACHE_DIR,
40
  )
41
 
42
  #TheBloke/Mistral-7B-v0.1-GGUF