Update app.py

app.py

@@ -36,7 +36,6 @@ def generate_text(prompt, max_length=100):
 
 # Hàm hiển thị thông tin API
 def get_api_info():
-    # Dùng placeholder, người dùng sẽ thay bằng URL thực tế của Space
     base_url = "https://<your-space-name>.hf.space"
     return (
         "Welcome to Qwen2.5-0.5B API!\n"
@@ -78,7 +77,7 @@ with gr.Blocks(title="Qwen2.5-0.5B Text Generator") as demo:
     )
 
 # Định nghĩa API endpoints với Gradio
-demo = gr.Interface(
+interface = gr.Interface(
     fn=generate_text,
     inputs=["text", "number"],
     outputs="text",
@@ -86,7 +85,6 @@ demo = gr.Interface(
     api_name="/generate"  # API endpoint: /api/generate
 ).queue()
 
-# Thêm endpoint health check
 health_interface = gr.Interface(
     fn=health_check,
     inputs=None,
@@ -94,8 +92,8 @@ health_interface = gr.Interface(
     api_name="/health_check"  # API endpoint: /api/health_check
 )
 
-# Kết hợp giao diện và API
-app = gr.mount_gradio_app(demo, health_interface)
+# Gắn các interface vào demo
+demo = gr.TabbedInterface([interface, health_interface], ["Generate Text", "Health Check"])
 
 # Chạy ứng dụng
 demo.launch(server_name="0.0.0.0", server_port=7860)