import gradio as gr

from transformers import pipeline
model = pipeline(task="question-answering", model="deepset/roberta-base-squad2")

context = """
The Sun is the star at the center of the Solar System. It is a nearly perfect ball of hot plasma, heated to incandescence by nuclear fusion reactions in its core, radiating the energy mainly as light, ultraviolet, and infrared radiation. It is the most important source of energy for life on Earth.

The Sun's diameter is about 1.39 million kilometers (864,000 miles), or 109 times that of Earth. Its mass is about 330,000 times that of Earth, comprising about 99.86% of the total mass of the Solar System. Roughly three-quarters of the Sun's mass consists of hydrogen (~73%); the rest is mostly helium (~25%), with much smaller quantities of heavier elements, including oxygen, carbon, neon, and iron.
"""

question = "What does the sun provide to the earth?"
response = model({ "question": question, "context": context })


def greet(name):
    response = model({ name: question, "context": context })
    return response

iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()