import solara
clicks = solara.reactive(0)
input = solara.reactive("I love transformers")
from transformers_js_py import import_transformers_js
model_id = 'Xenova/Phi-3-mini-4k-instruct_fp16'
@solara.lab.task
async def run(input):
transformers = await import_transformers_js()
pipeline = transformers.pipeline
pipe = await pipeline('text-generation', 'Xenova/distilgpt2')
out = await pipe('Once upon a time,')
return out
@solara.component
def Page():
if run.error:
solara.Error(repr(run.exception))
with solara.Card("Sentiment analysis"):
solara.ProgressLinear(run.pending)
with solara.Div():
solara.InputText(label="Input", value=input)
solara.Button(label=f"Analyze sentiment", on_click=lambda: run(input.value), color="primary", filled=True)
if run.finished:
solara.Text(repr(run.value))