Spaces:
Running
Running
region --> region_name
Browse files
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",
|
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
|