chenluuli commited on
Commit
8d85b7e
·
verified ·
1 Parent(s): 171f1eb

Update app.py

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