Fix compress only target

cornzz · Sep 9, 2024 · 748f122 · 748f122
1 parent cb6c9d4
commit 748f122
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/app.py b/src/app.py
@@ -180,7 +180,7 @@ def run_demo(
     if target_model == "Compress only":
         compressed, diff, metrics, compression_time = compress_prompt(prompt, rate, force_tokens, bool(force_digits))
         metrics["Compression"] = [f"{compression_time:.2f}s"]
-        return compressed, diff, metrics, None, None, None, None
+        return [compressed, diff, metrics] + [None] * 4 + [gr.Button(interactive=False)] * 3
 
     with ThreadPoolExecutor() as executor:
         get_query = lambda p: f"{question}\n\n{p}" if question else p