test

app.py

@@ -49,28 +49,28 @@ with gr.Blocks() as demo:
     clear = gr.ClearButton([prompt_POS, vicuna_S3_chatbot_POS])
     
     
-    gr.Markdown(" Description ")
+    # gr.Markdown(" Description ")
 
-    prompt_CHUNK = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
+    # prompt_CHUNK = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
 
-    gr.Markdown("Strategy 1 QA")
-    with gr.Row():
-        vicuna_S1_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
-        llama_S1_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
-        gpt_S1_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
-    clear = gr.ClearButton([prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
-    gr.Markdown("Strategy 2 Instruction")
-    with gr.Row():
-        vicuna_S2_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
-        llama_S2_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
-        gpt_S2_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
-    clear = gr.ClearButton([prompt_CHUNK, vicuna_S2_chatbot_CHUNK])
-    gr.Markdown("Strategy 3 Structured Prompting")
-    with gr.Row():
-        vicuna_S3_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
-        llama_S3_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
-        gpt_S3_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
-    clear = gr.ClearButton([prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
+    # gr.Markdown("Strategy 1 QA")
+    # with gr.Row():
+    #     vicuna_S1_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
+    #     llama_S1_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
+    #     gpt_S1_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
+    # clear = gr.ClearButton([prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
+    # gr.Markdown("Strategy 2 Instruction")
+    # with gr.Row():
+    #     vicuna_S2_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
+    #     llama_S2_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
+    #     gpt_S2_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
+    # clear = gr.ClearButton([prompt_CHUNK, vicuna_S2_chatbot_CHUNK])
+    # gr.Markdown("Strategy 3 Structured Prompting")
+    # with gr.Row():
+    #     vicuna_S3_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
+    #     llama_S3_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
+    #     gpt_S3_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
+    # clear = gr.ClearButton([prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
 
     # def respond(message, chat_history):
     #     input_ids = tokenizer.encode(message, return_tensors="pt")
@@ -90,14 +90,14 @@ with gr.Blocks() as demo:
         
         chat_history.append((message, bot_message))
         time.sleep(2)
-        return "", chat_history
+        return bot_message
 
     prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S1_chatbot_POS], [prompt_POS, vicuna_S1_chatbot_POS])
     prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S2_chatbot_POS], [prompt_POS, vicuna_S2_chatbot_POS])
     prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S3_chatbot_POS], [prompt_POS, vicuna_S3_chatbot_POS])
 
-    prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S1_chatbot_CHUNK], [prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
-    prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S2_chatbot_CHUNK], [prompt_CHUNK, vicuna_S2_chatbot_CHUNK])
-    prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S3_chatbot_CHUNK], [prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
+    # prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S1_chatbot_CHUNK], [prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
+    # prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S2_chatbot_CHUNK], [prompt_CHUNK, vicuna_S2_chatbot_CHUNK])
+    # prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S3_chatbot_CHUNK], [prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
 
 demo.launch()