change inference ui

app.py

@@ -62,7 +62,7 @@ def process_text(text: str):
 def inference(text):
     """Gradio interface function."""
     output, inference_time = process_text(text)
-    return f"Вихідний текст: {output}\nЧас обробки: {inference_time:.3f} секунд"
+    return output, f"{inference_time:.3f} секунд"
 
 # Define examples from inference_ct2.py
 examples = [
@@ -80,18 +80,29 @@ examples = [
 ]
 
 # Define Gradio interface
-interface = gr.Interface(
-    fn=inference,
-    inputs=gr.Textbox(
-        label="Введіть текст для вербалізації",
-        placeholder="Наприклад: Температура повітря сьогодні становить +25°C"
-    ),
-    outputs=gr.Textbox(label="Результат"),
-    examples=examples,
-    title="Вербалізація українського тексту (M2M100-CT2)",
-    description="""Модель для перетворення чисел, дат, одиниць вимірювання та інших символьних позначень 
-    у їх текстовий запис українською мовою. Використовує оптимізовану CTranslate2 версію для швидкого інференсу.""",
-    article="""
+with gr.Blocks() as interface:
+    gr.Markdown("# Вербалізація українського тексту (M2M100-CT2)")
+    gr.Markdown("""Модель для перетворення чисел, дат, одиниць вимірювання та інших символьних позначень 
+    у їх текстовий запис українською мовою. Використовує оптимізовану CTranslate2 версію для швидкого інференсу.""")
+    
+    with gr.Row():
+        with gr.Column():
+            input_text = gr.Textbox(
+                label="Введіть текст для вербалізації",
+                placeholder="Наприклад: Температура повітря сьогодні становить +25°C"
+            )
+            submit_btn = gr.Button("Вербалізувати")
+            
+            gr.Examples(
+                examples=examples,
+                inputs=input_text,
+            )
+            
+        with gr.Column():
+            output_text = gr.Textbox(label="Вербалізований текст")
+            inference_time = gr.Textbox(label="Час обробки")
+    
+    gr.Markdown("""
     ### Можливості моделі:
     - Дати та час
     - Телефонні номери
@@ -102,9 +113,13 @@ interface = gr.Interface(
     ### Технічні деталі:
     - Базова модель: facebook/m2m100_418M
     - Оптимізація: CTranslate2 з INT8 квантизацією
-    - Середній час інференсу: ~0.15-0.25 секунд на CPU
-    """
-)
+    """)
+    
+    submit_btn.click(
+        fn=inference,
+        inputs=input_text,
+        outputs=[output_text, inference_time],
+    )
 
 # Launch the interface
-interface.launch() 
+interface.launch(share=False)