CineAI commited on
Commit
06b83b7
1 Parent(s): 759a013

Update utils/statement_evaluation_command.py

Browse files
Files changed (1) hide show
  1. utils/statement_evaluation_command.py +11 -11
utils/statement_evaluation_command.py CHANGED
@@ -1,13 +1,13 @@
1
- import os
2
- from langchain_community.llms import HuggingFaceEndpoint
3
 
4
- _api = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
5
 
6
- def get_response(prompt: str):
7
- llm = HuggingFaceEndpoint(
8
- repo_id="CineAI/FreeThinker3B",
9
- max_length=1024,
10
- temperature=0.5,
11
- token=_api,
12
- )
13
- return llm.invoke(prompt)
 
1
+ # import os
2
+ # from langchain_community.llms import HuggingFaceEndpoint
3
 
4
+ # _api = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
5
 
6
+ # def get_response(prompt: str):
7
+ # llm = HuggingFaceEndpoint(
8
+ # repo_id="CineAI/FreeThinker3B",
9
+ # max_length=1024,
10
+ # temperature=0.5,
11
+ # token=_api,
12
+ # )
13
+ # return llm.invoke(prompt)