Spaces:
Sleeping
Sleeping
Update app_test.py
Browse files- 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 |
|