wishwakankanamg commited on
Commit
31030d9
·
1 Parent(s): 59a9b6b
Files changed (2) hide show
  1. README.md +1 -1
  2. agent.py +2 -1
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.py
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="togethercomputer/evo-1-131k-base",
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,