From 18fbf95f805d9334bc79e0f9e03d106e4188f4a6 Mon Sep 17 00:00:00 2001 From: Daniel Benner Date: Mon, 29 Apr 2024 17:47:22 -0500 Subject: [PATCH] updated to v2 of granite model as v1 is no longer supported --- discovery-data/webhook-enrichment-sample/granite/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/discovery-data/webhook-enrichment-sample/granite/main.py b/discovery-data/webhook-enrichment-sample/granite/main.py index 7d816c5..f288af9 100644 --- a/discovery-data/webhook-enrichment-sample/granite/main.py +++ b/discovery-data/webhook-enrichment-sample/granite/main.py @@ -40,7 +40,7 @@ def extract_entities(text): IAM_TOKEN = get_iam_token() # Prompt payload = { - 'model_id': 'ibm/granite-13b-instruct-v1', + 'model_id': 'ibm/granite-13b-instruct-v2', 'input': f'''Act as a webmaster who must extract structured information from emails. Read the below email and extract and categorize each entity. If no entity is found, output "None". Input: