brainsqueeze commited on
Commit
22b6a7a
·
verified ·
1 Parent(s): 6effa78

region --> region_name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def select_foundation_model(model_name: str, max_new_tokens: int) -> LLM:
38
  )
39
  elif model_name in {"claude-3.5-haiku", "llama-3.1-70b-instruct", "mistral-large", "mixtral-8x7B"}:
40
  llm = ChatBedrock(
41
- client=boto3.client("bedrock-runtime", region="us-east-1"),
42
  model=Name2Endpoint[model_name],
43
  max_tokens=max_new_tokens,
44
  temperature=0.0
 
38
  )
39
  elif model_name in {"claude-3.5-haiku", "llama-3.1-70b-instruct", "mistral-large", "mixtral-8x7B"}:
40
  llm = ChatBedrock(
41
+ client=boto3.client("bedrock-runtime", region_name="us-east-1"),
42
  model=Name2Endpoint[model_name],
43
  max_tokens=max_new_tokens,
44
  temperature=0.0