research14 commited on
Commit
2a20876
·
1 Parent(s): bbe45f1
Files changed (1) hide show
  1. app.py +24 -24
app.py CHANGED
@@ -49,28 +49,28 @@ with gr.Blocks() as demo:
49
  clear = gr.ClearButton([prompt_POS, vicuna_S3_chatbot_POS])
50
 
51
 
52
- gr.Markdown(" Description ")
53
 
54
- prompt_CHUNK = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
55
 
56
- gr.Markdown("Strategy 1 QA")
57
- with gr.Row():
58
- vicuna_S1_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
59
- llama_S1_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
60
- gpt_S1_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
61
- clear = gr.ClearButton([prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
62
- gr.Markdown("Strategy 2 Instruction")
63
- with gr.Row():
64
- vicuna_S2_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
65
- llama_S2_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
66
- gpt_S2_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
67
- clear = gr.ClearButton([prompt_CHUNK, vicuna_S2_chatbot_CHUNK])
68
- gr.Markdown("Strategy 3 Structured Prompting")
69
- with gr.Row():
70
- vicuna_S3_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
71
- llama_S3_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
72
- gpt_S3_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
73
- clear = gr.ClearButton([prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
74
 
75
  # def respond(message, chat_history):
76
  # input_ids = tokenizer.encode(message, return_tensors="pt")
@@ -90,14 +90,14 @@ with gr.Blocks() as demo:
90
 
91
  chat_history.append((message, bot_message))
92
  time.sleep(2)
93
- return "", chat_history
94
 
95
  prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S1_chatbot_POS], [prompt_POS, vicuna_S1_chatbot_POS])
96
  prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S2_chatbot_POS], [prompt_POS, vicuna_S2_chatbot_POS])
97
  prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S3_chatbot_POS], [prompt_POS, vicuna_S3_chatbot_POS])
98
 
99
- prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S1_chatbot_CHUNK], [prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
100
- prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S2_chatbot_CHUNK], [prompt_CHUNK, vicuna_S2_chatbot_CHUNK])
101
- prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S3_chatbot_CHUNK], [prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
102
 
103
  demo.launch()
 
49
  clear = gr.ClearButton([prompt_POS, vicuna_S3_chatbot_POS])
50
 
51
 
52
+ # gr.Markdown(" Description ")
53
 
54
+ # prompt_CHUNK = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
55
 
56
+ # gr.Markdown("Strategy 1 QA")
57
+ # with gr.Row():
58
+ # vicuna_S1_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
59
+ # llama_S1_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
60
+ # gpt_S1_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
61
+ # clear = gr.ClearButton([prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
62
+ # gr.Markdown("Strategy 2 Instruction")
63
+ # with gr.Row():
64
+ # vicuna_S2_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
65
+ # llama_S2_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
66
+ # gpt_S2_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
67
+ # clear = gr.ClearButton([prompt_CHUNK, vicuna_S2_chatbot_CHUNK])
68
+ # gr.Markdown("Strategy 3 Structured Prompting")
69
+ # with gr.Row():
70
+ # vicuna_S3_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
71
+ # llama_S3_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
72
+ # gpt_S3_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
73
+ # clear = gr.ClearButton([prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
74
 
75
  # def respond(message, chat_history):
76
  # input_ids = tokenizer.encode(message, return_tensors="pt")
 
90
 
91
  chat_history.append((message, bot_message))
92
  time.sleep(2)
93
+ return bot_message
94
 
95
  prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S1_chatbot_POS], [prompt_POS, vicuna_S1_chatbot_POS])
96
  prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S2_chatbot_POS], [prompt_POS, vicuna_S2_chatbot_POS])
97
  prompt_POS.submit(respond_entities, [prompt_POS, vicuna_S3_chatbot_POS], [prompt_POS, vicuna_S3_chatbot_POS])
98
 
99
+ # prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S1_chatbot_CHUNK], [prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
100
+ # prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S2_chatbot_CHUNK], [prompt_CHUNK, vicuna_S2_chatbot_CHUNK])
101
+ # prompt_CHUNK.submit(respond_entities, [prompt_CHUNK, vicuna_S3_chatbot_CHUNK], [prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
102
 
103
  demo.launch()