hqms commited on
Commit
799a14a
·
1 Parent(s): f7ccd51

add HF Key

Browse files
Files changed (2) hide show
  1. model/__init__.py +4 -0
  2. model/tools/__init__.py +4 -1
model/__init__.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
 
2
  label = {
3
  "emotion": ["sedih", "marah", "takut", "cinta", "senang", "netral"],
 
1
+ import os
2
+
3
+ token = os.environ.get("HF_KEY")
4
+
5
 
6
  label = {
7
  "emotion": ["sedih", "marah", "takut", "cinta", "senang", "netral"],
model/tools/__init__.py CHANGED
@@ -2,4 +2,7 @@ from langchain.chains import LLMChain
2
  from langchain.agents import Tool
3
  from langchain import HuggingFaceHub
4
 
5
- llm = HuggingFaceHub(repo_id = "microsoft/Phi-3-mini-128k-instruct")
 
 
 
 
2
  from langchain.agents import Tool
3
  from langchain import HuggingFaceHub
4
 
5
+ import os
6
+
7
+ hf_auth = os.environ.get("HF_KEY")
8
+ llm = HuggingFaceHub(repo_id = "microsoft/Phi-3-mini-128k-instruct", huggingfacehub_api_token = hf_auth)