app.py

import gradio as gr
from transformers import pipeline
import requests
import os
SECRET_TOKEN = os.getenv("SECRET_TOKEN")

API_URL = "https://api-inference.huggingface.co/models/foxxy-hm/mt5-small-finetuned-wikilingua-en-vi"
headers = {"Authorization": f"Bearer {SECRET_TOKEN}"}

def query(payload):
	# try:
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()[0]["generated_text"]
	# except Exception as e: # catch any kind of error
	# 	return f"Error: {e}" # return the error message


examples = [
    ["Cách phổ quát nhất để chào người khác bằng cử chỉ đó là bắt tay hoặc vẫy tay, đặc biệt là ở các quốc gia nói tiếng Anh. Tuy nhiên, một số cử chỉ khác như cúi đầu, ôm nhau, hoặc thậm chí là vỗ tay được sử dụng như là hành động chào hỏi ở các nước khác trên thế giới. Luôn bảo đảm rằng bạn không xúc phạm bất kỳ người nào có cách chào hỏi kỳ lạ ở một quốc gia nào đó"],
    ["Nếu tế bào da chết hoặc lớp biểu bì đổi màu do thuốc nhuộm, bạn có thể dùng dụng cụ cắt biểu bì để loại bỏ phần da này. Như vậy, bạn sẽ không cần phải dùng nước tẩy sơn móng tay có hại cho da. Nếu đã thử mọi cách mà móng vẫn dính thuốc nhuộm, tốt nhất bạn nên che màu thuốc nhuộm bằng màu sơn móng tay. Như vậy, bạn vừa trở nên sành điệu, vừa che được màu thuốc nhuộm tóc."],
    ["Nó là ứng dụng màu đen với chữ N màu đỏ. Gõ thông tin để đăng nhập nếu ứng dụng không tự động đăng nhập cho bạn. Bạn sẽ được kết nối tới đại diện của Trung tâm Trợ giúp Netflix. Ngoài ra, bạn có thể chạm vào Go to Help Center website (đi tới trang Trung tâm Trợ giúp) để tìm kiếm hoặc duyệt qua các chủ đề trợ giúp, hoặc tìm hiểu thêm về Netflix. Bạn sẽ được cung cấp một mã dịch vụ để sử dụng và khoảng thời gian chờ đợi dự kiến."],
    ["Dù không có vết bẩn trên thân quả thì bạn cũng nên rửa bí dưới vòi nước lạnh để làm sạch thuốc trừ sâu hoặc vi khuẩn sót lại. Dùng tay nhẹ nhàng xoa lớp vỏ bí ngòi, đồng thời rửa bí cho đến khi không còn cảm giác dính bẩn nữa. Dùng dao sắc bén cắt bỏ cuống và hoa bí ngòi. Ruột của những quả bí ngòi có đường kính lớn hơn 5cm thường sẽ lớn và đắng. Đối với những quả như vậy, bạn nên cẩn thận cắt đôi bí theo chiều dài, từ đầu này đến đầu kia. Một tay nắm chặt thìa, một tay cầm quả bí rồi ấn nhẹ đầu thìa từ trên xuống dưới để nạo sạch ruột bí. Nếu bí ngòi có đường kính nhỏ hơn thì không cần bỏ hạt trước khi bào."],
    ["Các viên dưa hấu lạnh là một món ăn tuyệt vời làm tỉnh người mà chắc chắn cả nhà bạn đều thích."],
    ["Đổ nước vào ấm và đun trên bếp với lửa vừa đến mạnh vừa cho đến khi sôi. Nhiều kiểu ấm trà có còi báo khi nước sôi, nhưng nếu ấm của bạn không có còi thì có lẽ bạn cần chịu khó để ý. Bạn cũng có thể dùng xoong nhỏ hoặc ấm điện để đun sôi nước. Lưu ý rằng có thể đun nước trong lò vi sóng, nhưng bạn nên đun sôi từng đợt ngắn khoảng 1-2 phút để tránh đun quá sôi. Bạn cũng nên đặt một chiếc đũa gỗ hoặc một vật dùng an toàn trong lò vi sóng vào nước khi đun"],
    ["Bạn có thể gửi ảnh và video có sẵn trong điện thoại của mình hoặc chụp và quay video nhanh sau đó gửi trực tiếp bằng Messenger. Tất cả các chức năng này đều được thực hiện trong cửa sổ hội thoại. Biểu tượng máy ảnh trên vùng soạn thảo văn bản cho phép bạn chụp ảnh và quay video sau đó gửi trực tiếp cho người nhận trong cuộc hội thoại. Khi sử dụng chức năng này lần đầu tiên, bạn có thể nhận được yêu cầu cho phép ứng dụng Messenger truy cập vào máy ảnh của thiết bị. Messenger chỉ thực hiện được chức năng này khi được bạn cấp quyền truy cập. Bạn có thể chuyển đổi giữa camera trước và sau bằng cách ấn vào biểu tượng ở góc dưới bên tay phải. Ấn nút \"Send\" (Gửi) để gửi ảnh vừa chụp vào cuộc hội thoại. Bạn có thể quay video với độ dài tối đa là 15 giây. Sau đó ấn nút \"Send\" (Gửi) để gửi video vào cuộc hội thoại. Bạn có thể hủy quay video bằng cánh di chuyển ngón tay ra khỏi nút chụp sau đó bỏ tay ra."],
]

demo = gr.Interface(
    fn=query,
    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
    outputs=gr.outputs.Textbox(label="Generated Text"),
    title="Text Summarization 📝", # The title of the interface
    description="This is a demo of text summarization for Vietnamese using mT5 fine-tuned model. The model performs better on shorter passages, typically consisting of 2-6 sentences.",
    examples=examples
)

demo.launch()