app.py

import gradio as gr
from transformers import pipeline
import spaces

# model = AutoModelForCausalLM.from_pretrained("Karzan/ckb-gpt2-medium-base-test-1024")
model_id = "Karzan/bart-qa-ckb"
pipe = pipeline(
    "question-answering",
    model=model_id,
    max_answer_len=300,
    handle_impossible_answer=True,
    top_k=3,
)
paragraph = """نوێترین توێژینەوەی پەیمانگەیەکی ئوسترالی ئاشکرای دەکات، لە ئەمریکا 60٪ـی ئەو خۆراکانەی بۆ منداڵانی تازەلەدایکبوو و ساوان، ناتەندروستن.
 
بەپێی توێژینەوەیەکی پەیمانگەی جۆرج بۆ تەندروستیی جیهانی، لەکۆی 651 خۆراکی بازرگانیی منداڵان، 70٪ـیان رێژەی ئاسایی پڕۆتینیان تێدا نەبووە و 44٪ـیان شەکری زیادیان تێدابووە.
 
ئێلیزابێس دانفۆرد، توێژەری سەرەکیی پەیمانگەکە دەڵێت: ''باشترین چارەسەر ئەوەیە کە نابێت منداڵ لە تەمەنێکی بچووکدا فێری خۆراکە ئامادەکراوەکان بکرێت و پێویستە بڕێکی کەم لەو خۆراکانە بدرێن بە منداڵ''.
 
ئێلیزابێس دانفۆرد دەشڵێت، زۆرینەی رێکلامەکانی خۆراکەکانی منداڵانی تازەلەدایکبوو چەواشەکارانەن، بەجۆرێک بانگەشەی بەرهەمی سرووشتی دەکەن لەنێو خۆراکەکانیان، بەڵام لە راستیدا بڕێکی زۆر 'نیشاستە، تام، مادەی پارێزەر و شەکر'ـیان تێدایە.
 
بەگوێرەی رێکخراوی تەندروستیی جیهانی، منداڵ دەبێت تاوەکو تەمەنی شەش مانگی تەنیا شیری دایک بخوات و تاوەکو تەمەنی دوو ساڵیش بەشێک لە خۆراکەکانی شیری دایک بێت، لەبەر ئەم هۆکارانەی خوارەوە؛"""

examples = [
    [
        paragraph,
        """لە سەدا چەندی خۆراکانەی ئەمەریکا تتەندروست نیین؟""",
    ],
    [
        paragraph,
       "کێ دەڵێت زۆرینەی رێکلامەکانی خۆراکەکانی منداڵانی تازەلەدایکبوو چەواشەکارانەن؟",
    ],
]


@spaces.GPU(duration=120)
def func(context, question):
    result = pipe(question=question, context=context)
    return (
        result[0]["answer"],
        result[0]["score"],
        result[1]["answer"],
        result[1]["score"],
        result[2]["answer"],
        result[2]["score"],
    )


interface = gr.Interface(
    fn=func,
    inputs=[
        gr.Textbox(lines=7, label="Context paragraph"),
        gr.Textbox(lines=2, label="Question"),
    ],
    outputs=[
        gr.Textbox(label="Answer"),
        gr.Textbox(label="Score"),
        gr.Textbox(label="Answer"),
        gr.Textbox(label="Score"),
        gr.Textbox(label="Answer"),
        gr.Textbox(label="Score"),
    ],
    examples=examples,
    # submit_btn=submit_btn,
)

if __name__ == "__main__":
    interface.launch()