wenhu commited on
Commit
c2ad9fa
ยท
verified ยท
1 Parent(s): 65d242b

Update app_test.py

Browse files
Files changed (1) hide show
  1. app_test.py +47 -0
app_test.py CHANGED
@@ -63,6 +63,53 @@ external_log_dir = "./logs"
63
  LOGDIR = external_log_dir
64
  VOTEDIR = "./votes"
65
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
66
  if __name__ == "__main__":
67
  import argparse
68
 
 
63
  LOGDIR = external_log_dir
64
  VOTEDIR = "./votes"
65
 
66
+
67
+ with gr.Blocks(
68
+ css=".message-wrap.svelte-1lcyrx4>div.svelte-1lcyrx4 img {min-width: 40px}",
69
+ ) as demo:
70
+ cur_dir = os.path.dirname(os.path.abspath(__file__))
71
+ # gr.Markdown(title_markdown)
72
+ gr.HTML(html_header)
73
+
74
+ with gr.Column():
75
+ with gr.Accordion("Parameters", open=False) as parameter_row:
76
+ temperature = gr.Slider(
77
+ minimum=0.0,
78
+ maximum=1.0,
79
+ value=0.7,
80
+ step=0.1,
81
+ interactive=True,
82
+ label="Temperature",
83
+ )
84
+ top_p = gr.Slider(
85
+ minimum=0.0,
86
+ maximum=1.0,
87
+ value=1,
88
+ step=0.1,
89
+ interactive=True,
90
+ label="Top P",
91
+ )
92
+ max_output_tokens = gr.Slider(
93
+ minimum=0,
94
+ maximum=8192,
95
+ value=4096,
96
+ step=256,
97
+ interactive=True,
98
+ label="Max output tokens",
99
+ )
100
+ with gr.Row():
101
+ chatbot = gr.Chatbot([], elem_id="MAmmoTH-VL-8B", bubble_full_width=False, height=750)
102
+
103
+ with gr.Row():
104
+ upvote_btn = gr.Button(value="๐Ÿ‘ Upvote", interactive=True)
105
+ downvote_btn = gr.Button(value="๐Ÿ‘Ž Downvote", interactive=True)
106
+ flag_btn = gr.Button(value="โš ๏ธ Flag", interactive=True)
107
+ # stop_btn = gr.Button(value="โน๏ธ Stop Generation", interactive=True)
108
+ regenerate_btn = gr.Button(value="๐Ÿ”„ Regenerate", interactive=True)
109
+ clear_btn = gr.Button(value="๐Ÿ—‘๏ธ Clear history", interactive=True)
110
+
111
+ demo.queue()
112
+
113
  if __name__ == "__main__":
114
  import argparse
115