Spaces:

chansung
/

llama2-with-gradio-chat

Running on T4

chansung commited on Aug 16, 2023

Commit

c205d24

1 Parent(s): 13272b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -88,7 +88,7 @@ async def chat_stream(
         }
     ):
         ppm.append_pong(result)
-        yield ppm.build_uis(), str(res)
 def channel_num(btn_title):
     choice = 0
@@ -247,7 +247,7 @@ with gr.Blocks(css=MODEL_SELECTION_CSS, theme='gradio/soft') as demo:
         chat_stream,
         [idx, local_data, instruction_txtbox, chat_state,
          global_context, res_temp, res_topk, res_rpen, res_mnts, res_sample, ctx_num_lconv],
-        [chatbot, local_data]
     )
     for btn in channel_btns:

         }
     ):
         ppm.append_pong(result)
+        yield prompt, ppm.build_uis(), str(res)
 def channel_num(btn_title):
     choice = 0
         chat_stream,
         [idx, local_data, instruction_txtbox, chat_state,
          global_context, res_temp, res_topk, res_rpen, res_mnts, res_sample, ctx_num_lconv],
+        [context_inspector, chatbot, local_data]
     )
     for btn in channel_btns: