add HF Key
Browse files- model/__init__.py +4 -0
 - 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 | 
         
            -
             
     | 
| 
         | 
|
| 
         | 
|
| 
         | 
| 
         | 
|
| 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)
         
     |