mirror of
https://github.com/gradio-app/gradio.git
synced 2025-02-11 11:19:58 +08:00
12 lines
363 B
Python
12 lines
363 B
Python
|
import transformers
|
||
|
|
|
||
|
|
import gradio as gr
|
||
|
|
|
||
|
|
|
||
|
|
class TestLoadFromPipeline:
|
||
|
|
def test_text_to_text_model_from_pipeline(self):
|
||
|
|
pipe = transformers.pipeline(model="sshleifer/bart-tiny-random")
|
||
|
|
io = gr.Interface.from_pipeline(pipe)
|
||
|
|
output = io("My name is Sylvain and I work at Hugging Face in Brooklyn")
|
||
|
|
assert isinstance(output, str)
|