Commit
·
31030d9
1
Parent(s):
59a9b6b
sdfsdf
Browse files
README.md
CHANGED
@@ -5,7 +5,7 @@ colorFrom: indigo
|
|
5 |
colorTo: indigo
|
6 |
sdk: gradio
|
7 |
sdk_version: 5.25.2
|
8 |
-
app_file: agent
|
9 |
pinned: false
|
10 |
hf_oauth: true
|
11 |
# optional, default duration is 8 hours/480 minutes. Max duration is 30 days/43200 minutes.
|
|
|
5 |
colorTo: indigo
|
6 |
sdk: gradio
|
7 |
sdk_version: 5.25.2
|
8 |
+
app_file: agent..py
|
9 |
pinned: false
|
10 |
hf_oauth: true
|
11 |
# optional, default duration is 8 hours/480 minutes. Max duration is 30 days/43200 minutes.
|
agent.py
CHANGED
@@ -167,10 +167,11 @@ def build_graph(provider: str = "huggingface"):
|
|
167 |
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
168 |
elif provider == "huggingface":
|
169 |
repo_id = "togethercomputer/evo-1-131k-base"
|
|
|
170 |
if not hf_token:
|
171 |
raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
|
172 |
llm = HuggingFaceEndpoint(
|
173 |
-
repo_id="
|
174 |
task="text-generation",
|
175 |
max_new_tokens=1000,
|
176 |
do_sample=False,
|
|
|
167 |
llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
|
168 |
elif provider == "huggingface":
|
169 |
repo_id = "togethercomputer/evo-1-131k-base"
|
170 |
+
repo_id="HuggingFaceH4/zephyr-7b-beta",
|
171 |
if not hf_token:
|
172 |
raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
|
173 |
llm = HuggingFaceEndpoint(
|
174 |
+
repo_id="sentence-transformers/all-MiniLM-L6-v2",
|
175 |
task="text-generation",
|
176 |
max_new_tokens=1000,
|
177 |
do_sample=False,
|