Change default base model to mbert-base due to limited computation.

app.py

@@ -35,7 +35,7 @@ def compress(original_prompt, compression_rate, base_model="xlm-roberta-large",
         idx = force_tokens.index('\\n')
         force_tokens[idx] = '\n'
 
-    compressor = compressors.get(base_model, compressors["xlm-roberta-large"])
+    compressor = compressors.get(base_model, compressors["mbert-base"])
     results = compressor.compress_prompt_llmlingua2(
             original_prompt,
             rate=compression_rate,
@@ -86,7 +86,7 @@ with gr.Blocks(title=title, css=css) as app: # 'YenLai/Superhuman' 'HaleyCH/Hale
             compressed_prompt = gr.Textbox(value='', label="Compressed Prompt", lines=10, max_lines=10, interactive=False)
             
         with gr.Column(scale=1):
-            base_model = gr.Radio(["xlm-roberta-large", "mbert-base"], label="Base Model", value="xlm-roberta-large", interactive=True)
+            base_model = gr.Radio(["mbert-base", "xlm-roberta-large"], label="Base Model", value="mbert-base", interactive=True)
             force_tokens = gr.Dropdown(['\\n', '.', '!', '?', ','],
                                        label="Tokens to Preserve",
                                        value=['\\n', '.', '!', '?', ','],