ikun520 commited on
Commit
a3dd631
·
verified ·
1 Parent(s): 5180982

Update model_utils.py

Browse files
Files changed (1) hide show
  1. model_utils.py +1 -1
model_utils.py CHANGED
@@ -32,7 +32,7 @@ def matter(txt):
32
  spark = get_chat_model()
33
  messages = [ChatMessage(
34
  role="user",
35
- content=f'你是一个图谱实体知识结构化专家。根据输入实体类型(entity type)的schema描述,从文本中抽取出相应的实体实例和其属性信息,不存在的属性不输出, 属性存在多值就返回列表,并输出为可解析的json格式。不需要输出代码,{txt}\n'
36
  )]
37
  a = spark.generate([messages])
38
  return a.generations[0][0].text
 
32
  spark = get_chat_model()
33
  messages = [ChatMessage(
34
  role="user",
35
+ content=txt
36
  )]
37
  a = spark.generate([messages])
38
  return a.generations[0][0].text