Update config.json

config.json

@@ -1,17 +1,17 @@
 {
- "model_name": "slim-extract-tool",
- "model_ft_base": "slim-extract",
+ "model_name": "slim-sa-ner-tool",
+ "model_ft_base": "slim-sa-ner",
  "quantization": "4Q_K_M GGUF",
  "model_base": "stabilityai/stablelm-3b-4e1t",
  "model_type": "stablelm",
- "tokenizer": "llmware/slim-extract",
+ "tokenizer": "llmware/slim-sa-ner",
  "parameters": "2.7 billion",
- "description": "slim-extract is a function-calling model, fine-tuned to output structured dictionaries corresponding to a custom extraction key",
+ "description": "slim-sa-ner is a function-calling model, fine-tuned to output structured dictionaries corresponding to sentiment and NER keys",
  "prompt_wrapper": "human_bot",
- "prompt_format": "<human> {context_passage} <classify> {custom extraction key} </classify>\n<bot>:",
- "output_format": "{custom_extraction_key: ['list of items found in the text corresponding to key']}",
- "primary_keys": ["any custom key provided"],
- "output_values": ["dictionary with custom_key and value consisting of list of extracted values"],
+ "prompt_format": "<human> {context_passage} <classify> sentiment, people, organization, place </classify>\n<bot>:",
+ "output_format": "{dictionary consisting of sentiment, people, organization, and place keys}",
+ "primary_keys": ["sentiment, people, organization, place"],
+ "output_values": ["each key will have a value consisting of a list"],
  "publisher": "llmware",
  "release_date": "march 2024",
  "test_set": [