Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ def format_prompt(message, history):
|
|
| 26 |
|
| 27 |
def chat_inf(system_prompt,prompt,history,client_choice):
|
| 28 |
#token max=8192
|
| 29 |
-
client=clients[int(client_choice)]
|
| 30 |
if not history:
|
| 31 |
history = []
|
| 32 |
hist_len=0
|
|
@@ -43,8 +43,8 @@ def chat_inf(system_prompt,prompt,history,client_choice):
|
|
| 43 |
do_sample=True,
|
| 44 |
seed=seed,
|
| 45 |
)
|
| 46 |
-
|
| 47 |
-
formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
|
| 48 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
| 49 |
output = ""
|
| 50 |
|
|
@@ -68,5 +68,7 @@ with gr.Blocks() as app:
|
|
| 68 |
stop_btn=gr.Button("Stop")
|
| 69 |
clear_btn=gr.Button("Clear")
|
| 70 |
client_choice=gr.Dropdown(label="Models",type='index',choices=[c for c in models],value=models[0],interactive=True)
|
| 71 |
-
btn.click(chat_inf,[sys_inp,inp,chat_b,client_choice],chat_b)
|
|
|
|
|
|
|
| 72 |
app.launch()
|
|
|
|
| 26 |
|
| 27 |
def chat_inf(system_prompt,prompt,history,client_choice):
|
| 28 |
#token max=8192
|
| 29 |
+
client=clients[int(client_choice)-1]
|
| 30 |
if not history:
|
| 31 |
history = []
|
| 32 |
hist_len=0
|
|
|
|
| 43 |
do_sample=True,
|
| 44 |
seed=seed,
|
| 45 |
)
|
| 46 |
+
formatted_prompt=prompt
|
| 47 |
+
#formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
|
| 48 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
| 49 |
output = ""
|
| 50 |
|
|
|
|
| 68 |
stop_btn=gr.Button("Stop")
|
| 69 |
clear_btn=gr.Button("Clear")
|
| 70 |
client_choice=gr.Dropdown(label="Models",type='index',choices=[c for c in models],value=models[0],interactive=True)
|
| 71 |
+
go=btn.click(chat_inf,[sys_inp,inp,chat_b,client_choice],chat_b)
|
| 72 |
+
stop_btn.click(None,None,None,cancels=go)
|
| 73 |
+
|
| 74 |
app.launch()
|