Update UI/gradio.py

UI/gradio.py

@@ -295,12 +295,12 @@ class WatermarkerInterface:
 
         print("Reparaphrased indices:", self.reparaphrased_indices)
 
-        while len(html_outputs) < 120:
+        while len(html_outputs) < 150:
             html_outputs.append("")
 
         execution_time = f"<div class='execution-time'>Step 4 completed in {time.time() - start_time:.2f} seconds</div>"
 
-        return html_outputs[:120] + [execution_time]
+        return html_outputs[:150] + [execution_time]
     
 def create_gradio_interface(config):
     """Creates the Gradio interface with the updated pipeline"""
@@ -530,7 +530,7 @@ def create_gradio_interface(config):
         gr.Markdown("### Reparaphrased Sentences")
         reparaphrased_sentences_tabs = []
         with gr.Tabs(elem_id="reparaphrased-tabs") as reparaphrased_tabs:
-            for i in range(120):
+            for i in range(150):
                 with gr.TabItem(f"Reparaphrased Batch {i+1}"):
                     reparaphrased_sent_html = gr.HTML()
                     reparaphrased_sentences_tabs.append(reparaphrased_sent_html)