Update app.py

app.py

@@ -68,7 +68,7 @@ def demo():
 
     translation_mode_map = {
         "translate": "exact",
-        "translate+extend": "approx",
+        "translate + extend": "approx",
     }
 
     @spaces.GPU(duration=5)
@@ -115,7 +115,7 @@ def demo():
             bad_words_ids=ban_ids,
         )
         elapsed = time.time() - start_time
-
+        
         deocded = ", ".join(
             [
                 tag
@@ -191,8 +191,8 @@ def demo():
                         )
 
                 with gr.Column():
-                    output_translation = gr.Textbox(label="Output (translation)", lines=4, interactive=False)
-                    output_extension = gr.Textbox(label="Output (extension)", lines=4, interactive=False)
+                    output_translation = gr.Textbox(label="Output", lines=4, interactive=False)
+                    # output_extension = gr.Textbox(label="Output (extension)", lines=4, interactive=False)
                     time_elapsed = gr.Markdown(value="")
 
             gr.Examples(
@@ -266,7 +266,11 @@ def demo():
                 top_k,
                 top_p,
             ],
-            outputs=[output_translation, output_extension, time_elapsed],
+            outputs=[
+                output_translation, 
+                # output_extension, 
+                time_elapsed,
+            ],
         )
 
     ui.launch()