Spaces:

sunny77
/

Retrieval-Augmented-Generation-RAG

Runtime error

sunny77 commited on Dec 23, 2023

Commit

9e72591

1 Parent(s): 02d2c87

env

Files changed (1) hide show

retrieve.py CHANGED Viewed

@@ -28,6 +28,7 @@ def load_vector_store():
 #model
 def load_model():
     load_dotenv()
     repo_id = 'llmware/bling-sheared-llama-1.3b-0.1'
     llm = HuggingFaceHub(
         repo_id = repo_id,

 #model
 def load_model():
     load_dotenv()
+    os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.getenv("HUGGINGFACEHUB_API_TOKEN")
     repo_id = 'llmware/bling-sheared-llama-1.3b-0.1'
     llm = HuggingFaceHub(
         repo_id = repo_id,