Merge branch 'main' of https://huggingface.co/spaces/nvidia/P2A-test-NV

app.py

@@ -90,8 +90,10 @@ def check_token_length(text, max_tokens=1024):
     return len(text) <= max_tokens
 
 import uuid
-def get_user_session(state):
-    if not state:
+def get_user_session(state=None):
+    if state is None:
+        state = {}
+    if not isinstance(state, dict):
         state = {}
     if not state.get("session_id"):
         state["session_id"] = uuid.uuid4().hex
@@ -213,7 +215,6 @@ def process_text(text, src_language, target_language, max_iterations_value, thre
                  good_ref_contexts_num_value, translation_methods=None, state=None):
     
     translation_methods = translation_methods or ["Original", "Plan2Align"]
-    state = state or {}
     session_id = get_user_session(state)
 
     """
@@ -325,11 +326,11 @@ with gr.Blocks(title="Test-Time Machine Translation with Plan2Align") as demo:
     
     gr.Examples(
         examples=[
-            ["台灣夜市文化豐富多彩，從士林夜市到饒河街夜市，提供各種美食、遊戲和購物體驗，吸引了無數遊客。", "Chinese", "English", 2, 0.7, 1],
-            ["台北101曾經是世界最高的建築物，它不僅是台灣的地標，也象徵著經濟成就和創新精神。", "Chinese", "Russian", 2, 0.7, 1],
-            ["阿里山日出和森林鐵路是台灣最著名的自然景觀之一，每年吸引數十萬遊客前來欣賞雲海和壯麗的日出。", "Chinese", "German", 2, 0.7, 1],
-            ["珍珠奶茶，這款源自台灣的獨特飲品，不僅在台灣本地深受喜愛，更以其獨特的風味和口感，在全球掀起了一股熱潮，成為了一種跨越文化、風靡全球的時尚飲品。", "Chinese", "Japanese", 3, 0.7, 3],
-            ["原住民文化如同一片深邃的星空，閃爍著無數璀璨的傳統與藝術光芒。他們的歌舞，是與祖靈對話的旋律，是與自然共鳴的節奏，每一個舞步、每一聲吟唱，都承載著古老的傳說與智慧。編織，是他們巧手下的詩篇，一絲一線，交織出生命的紋理，也編織出對土地的熱愛與敬畏。木雕，則是他們與自然對話的雕塑，每一刀、每一鑿，都刻畫著對萬物的觀察與敬意，也雕琢出對祖先的追憶與傳承。", "Chinese", "Korean", 5, 0.7, 5]
+            ["台灣夜市文化豐富多彩...", "Chinese", "English", 2, 0.7, 1, ["Original", "Plan2Align"]],
+            ["台北101曾經是世界最高的建築物，它不僅是台灣的地標，也象徵著經濟成就和創新精神。", "Chinese", "Russian", 2, 0.7, 1, ["Original", "Plan2Align"]],
+            ["阿里山日出和森林鐵路是台灣最著名的自然景觀之一，每年吸引數十萬遊客前來欣賞雲海和壯麗的日出。", "Chinese", "German", 2, 0.7, 1, ["Original", "Plan2Align"]],
+            ["珍珠奶茶，這款源自台灣的獨特飲品，不僅在台灣本地深受喜愛，更以其獨特的風味和口感，在全球掀起了一股熱潮，成為了一種跨越文化、風靡全球的時尚飲品。", "Chinese", "Japanese", 3, 0.7, 3, ["Original", "Plan2Align"]],
+            ["原住民文化如同一片深邃的星空，閃爍著無數璀璨的傳統與藝術光芒。他們的歌舞，是與祖靈對話的旋律，是與自然共鳴的節奏，每一個舞步、每一聲吟唱，都承載著古老的傳說與智慧。編織，是他們巧手下的詩篇，一絲一線，交織出生命的紋理，也編織出對土地的熱愛與敬畏。木雕，則是他們與自然對話的雕塑，每一刀、每一鑿，都刻畫著對萬物的觀察與敬意，也雕琢出對祖先的追憶與傳承。", "Chinese", "Korean", 5, 0.7, 5, ["Original", "Plan2Align"]]
         ],
         inputs=[
             source_text,
@@ -337,7 +338,8 @@ with gr.Blocks(title="Test-Time Machine Translation with Plan2Align") as demo:
             task_language_input,
             max_iterations_input,
             threshold_input,
-            good_ref_contexts_num_input
+            good_ref_contexts_num_input,
+            translation_methods_input
         ],
         outputs=[original_output, plan2align_output, best_of_n_output, mpc_output],
         fn=process_text
@@ -354,4 +356,4 @@ with gr.Blocks(title="Test-Time Machine Translation with Plan2Align") as demo:
     """)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)