Ngadou's picture
Update app.py
f9f149a verified
raw
history blame
2.65 kB
from typing import List , Tuple , Union
from web_ui import WebUI
import math
import os
from qwen_agent.agents import Assistant
from qwen_agent.gui.gradio import gr
def app_gui ():
# Define the agent
bot = Assistant(llm={
'model' : os.environ.get( "MODELNAME" ),
'model_type' : 'qwen_dashscope' ,
'generate_cfg' : {
'max_input_tokens' : 32768 ,
'max_retries' : 10 ,
'temperature' : float (os.environ.get( "T" , 0.001 )),
'repetition_penalty' : float (os.environ.get( "R" , 1.0 )),
"top_k" : int (os.environ.get( "K" , 20 )),
"top_p" : float (os.environ.get( "P" , 0.8 )),
}},
name = 'QwQ-32B-preview' ,
description= 'QwQ-32B-Preview is an experimental research model developed by the Qwen Team, focused on advancing AI reasoning capabilities. As a preview release, it demonstrates promising analytical abilities while having several important limitations such as code switching and recursive reasoning loops. Only single-turn queries are supported in this demo.' ,
system_message= 'You are a helpful and harmless assistant. You are Qwen developed by Alibaba. You should think step-by-step.' ,
rag_cfg={ 'max_ref_token' : 32768 , 'rag_searchers' : []},
)
chatbot_config = {
'input.placeholder' : "Type \"/clear\" to clear the history" ,
'verbose' : True ,
'prompt.suggestions' : [
{
'text' : 'How many r in strawberry'
},
{
'text' : 'Find the least odd prime factor of $2019^8+1$.'
},
{
'text' : '''Mr. S, Mr. P, and Mr. Q know that there are 16 playing cards in the drawer of the table: A, Q, 4 of hearts, J, 8, 4, 2, 7, 3 of spades, K, Q, 5, 4, 6 of clubs, A, 5 of diamonds. Professor John picked a card from the 16 cards and told Mr. P the number of points on the card and Mr. Q the suit of the card. At this time, Professor John asked Mr. P and Mr. Q: Can you infer what card this is from the known number of points or suit? So, Mr. S heard the following conversation:
Mr. P: I don’t know this card.
Mr. Q: I know you don’t know this card.
Mr. P: Now I know this card.
Mr. Q: I know that too.
Please tell me: What card is this? '''
},
]
}
WebUI(bot, chatbot_config=chatbot_config).run(concurrency_limit= 80)
if __name__ == '__main__' :
app_gui()