Upload app.py

app.py

@@ -17,12 +17,12 @@ tokenizer = tiktoken.encoding_for_model("gpt-4")
 with open('data/benchmark_33_bctn_so_lieu_5context.json', 'r') as f:
     examples = json.load(f)
 
-def compress(original_prompt, compression_rate, base_model="xlm-roberta-large", force_tokens=['\n'], chunk_end_tokens=['.', '\n']):
+def compress(original_prompt, compression_rate, base_model="xlm-roberta", force_tokens=['\n'], chunk_end_tokens=['.', '\n']):
     if '\\n' in force_tokens:
         idx = force_tokens.index('\\n')
         force_tokens[idx] = '\n'
 
-    compressor = compressors.get(base_model, compressors["mbert-base"])
+    compressor = compressors.get(base_model, compressors["xlm-roberta"])
     results = compressor.compress_prompt_llmlingua2(
             original_prompt,
             rate=compression_rate,
@@ -54,8 +54,7 @@ theme = "soft"
 css = """#anno-img .mask {opacity: 0.5; transition: all 0.2s ease-in-out;}
             #anno-img .mask.active {opacity: 0.7}"""
 
-original_prompt_text = """
-"""
+original_prompt_text = """"""
 
 with gr.Blocks(title=title, css=css) as app:
     gr.Markdown(header)