Feature(LLMLingua): build the LLMLingua space demo

app.py

@@ -47,13 +47,10 @@ custom_css = """
     }
 """
 
-def compress_prompt(context, instruct, question, ratio, target_token):
+def compress_prompt(context, instruction, question, ratio, target_token):
     compressed_prompt = llm_lingua.compress_prompt(context.split("\n\n"), instruction, question, ratio, target_token)
 
-    return [compressed_prompt[key] for key in ["compressed_prompt", "origin_tokens", "compressed_tokens", "ratio", "saving"]]
-
-iface = gr.Interface(fn=compress_prompt, inputs="text", outputs="text")
-iface.launch()
+    return *[compressed_prompt[key] for key in ["compressed_prompt", "origin_tokens", "compressed_tokens", "ratio", "saving"]]
 
 
 with gr.Blocks(css=custom_css) as iface: