lchakkei
/

2

Text Generation

text-generation-inference

Model card Files Files and versions

lchakkei commited on Feb 6, 2024

Commit

ad72be6

·

verified ·

1 Parent(s): b4768bf

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -30,8 +30,8 @@ class EndpointHandler():
     def __init__(self, path=""):
         # Config LangChain
-        # os.environ["LANGCHAIN_TRACING_V2"] = "true"
-        # os.environ["LANGCHAIN_API_KEY"] = "ls__9834e6b2ff094d43a28418c9ecea2fd5"
         # Create LLM
         model_id = path

     def __init__(self, path=""):
         # Config LangChain
+        os.environ["LANGCHAIN_TRACING_V2"] = "true"
+        os.environ["LANGCHAIN_API_KEY"] = "ls__9834e6b2ff094d43a28418c9ecea2fd5"
         # Create LLM
         model_id = path