chenluuli commited on
Commit
171f1eb
·
verified ·
1 Parent(s): 7f2032c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -35,7 +35,11 @@ def initClient():
35
  def greet(name):
36
  modelName = "chenluuli/test-text-vis"
37
  text2text_generator = pipeline("text2text-generation", model="Qwen/Qwen2.5-0.5B-Instruct")
38
- response = text2text_generator("##你是一个可视化专家,通过我提供的信息,推荐合理的图表配置###我提供的信息可能有,字段信息metas,明细数据data,分析意图purpose,用户输入input!!请根据这些信息,返回合理的图表配置,不需要额外的文字描述 >>我输入的数据如下:"+name)
 
 
 
 
39
  print(response, response[0]['generated_text'])
40
  return response[0]['generated_text']
41
 
 
35
  def greet(name):
36
  modelName = "chenluuli/test-text-vis"
37
  text2text_generator = pipeline("text2text-generation", model="Qwen/Qwen2.5-0.5B-Instruct")
38
+ prompt = "##你是一个可视化专家,通过我提供的信息,推荐合理的图表配置###我提供的信息可能有,字段信息metas,明细数据data,分析意图purpose,用户输入input!!请根据这些信息,返回合理的图表配置,不需要额外的文字描述 >>我输入的数据如下:"
39
+ response = text2text_generator(
40
+ prompt+name,
41
+ max_new_tokens=512
42
+ )
43
  print(response, response[0]['generated_text'])
44
  return response[0]['generated_text']
45