windy/streamlit-testGenius
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

format code

Browse Source
This commit is contained in:
VincentXiuyuanZhao 2024-06-07 18:55:57 +08:00
parent 6fec5098d4
commit ba969a699e
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
simpleGradio.py
View File

@ -9,19 +9,25 @@ import os
dashscope.api_key = os.getenv("DASHSCOPE_API_KEY") # Vincent's API key dashscope.api_key = os.getenv("DASHSCOPE_API_KEY") # Vincent's API key
# todo: add question variable
def response(prompt, instruction=None): def response(prompt, temp, question=None):
# if instruction is not None: # 如果提供了指令，则添加到messages中
# messages.insert(0, {'role': 'system', 'content': instruction})
if question is not None: # 如果提供了问题，则添加到messages中
# if no {question}, raise error
if '{question}' not in prompt:
raise ValueError('Please provide a question variable')
prompt = prompt.replace('{question}', question)
else:
pass
messages = [{'role': 'user', 'content': prompt}] messages = [{'role': 'user', 'content': prompt}]
if instruction is not None: # 如果提供了指令，则添加到messages中
messages.insert(0, {'role': 'system', 'content': instruction})
response = Generation.call(model='qwen-plus', response = Generation.call(model='qwen-plus',
messages=messages, messages=messages,
seed=1234, seed=1234,
result_format='message', result_format='message',
stream=False, stream=False,
incremental_output=False, incremental_output=False,
temperature=1.8, temperature=temp,
top_p=0.9, top_p=0.9,
top_k=999 top_k=999
) )
@ -37,14 +43,15 @@ def response(prompt, instruction=None):
with gr.Blocks() as demo: with gr.Blocks() as demo:
input_text = gr.Textbox(label="输入文本") input_text = gr.Textbox(label="prompt")
question_text = gr.Textbox(label="question")
output_text = gr.Textbox(label="输出文本") output_text = gr.Textbox(label="输出文本")
submit_button = gr.Button("submit") submit_button = gr.Button("submit")
# 使用 gr.ClearButton 来清除指定的输出组件 # 使用 gr.ClearButton 来清除指定的输出组件
clear_button = gr.ClearButton(components=[output_text], value="Clear processed data") clear_button = gr.ClearButton(components=[output_text], value="Clear processed data")
temp = gr.Slider(label="Temperature", minimum=0.1, maximum=2.0, step=0.05, value=1.8)
# 当输入文本发生变化时，调用 response 函数并将结果显示在输出文本框中 # 当输入文本发生变化时，调用 response 函数并将结果显示在输出文本框中
submit_button.click(fn=response, inputs=input_text, outputs=output_text) submit_button.click(fn=response, inputs=[input_text, temp, question_text], outputs=output_text)
demo.launch() demo.launch()