Feature(LLMLingua): changing the style

app.py

@@ -52,7 +52,6 @@ custom_css = """
 
 def compress_prompt(context, instruction, question, ratio, target_token):
     context, instruction, question = context.replace("\\n", "\n"), instruction.replace("\\n", "\n"), question.replace("\\n", "\n")
-    print(context)
     compressed_prompt = llm_lingua.compress_prompt(context.split("\n\n"), instruction, question, float(ratio), float(target_token))
 
     return [compressed_prompt[key] for key in ["compressed_prompt", "origin_tokens", "compressed_tokens", "ratio", "saving"]]
@@ -62,7 +61,7 @@ with gr.Blocks(css=custom_css) as iface:
     gr.Markdown(INTRO)
 
     with gr.Row():
-        with gr.Column(elem_id="params"):
+        with gr.Column(elem_id="prompt"):
             with gr.Tab('Prompts'):
                 instruction = gr.Textbox(
                     label="Instruction",
@@ -79,25 +78,19 @@ with gr.Blocks(css=custom_css) as iface:
                     lines=1,
                     value="",
                 )
+        with gr.Column(elem_id="params"):
             with gr.Tab('Compression Target'):
-                ratio = gr.Textbox(
-                    label="Compression Ratio",
-                    value=0.5,
-                )
                 target_token = gr.Textbox(
-                    label="Target Token",
-                    value=-1,
+                    label="Target Token (To use this, set Compression Ratio to 0)",
+                    value=200,
+                )
+                ratio = gr.Textbox(
+                    label="Compression Ratio (To use this, set Target Token to -1)",
+                    value=0,
                 )
 
     gen_button = gr.Button(value="Compress Prompt!", variant="primary")
 
-    # with gr.Tab('Results'):
-    #     results = gr.Gallery(
-    #         show_label=False,
-    #         object_fit="contain",
-    #         columns=4
-    #     )
-
     with gr.Row():
         with gr.Column(elem_id="Results"):
             with gr.Tab('Compressed Prompts'):
@@ -105,6 +98,8 @@ with gr.Blocks(css=custom_css) as iface:
                     label="compressed_prompt",
                     lines=5,
                 )
+    with gr.Row():
+        with gr.Column(elem_id="Results_2"):
             with gr.Tab('Saving'):
                 origin_tokens = gr.Textbox(
                     label="The tokens number of original prompt",