mirror of
https://github.com/gradio-app/gradio.git
synced 2025-01-06 10:25:17 +08:00
87f1c2b4ac
* add test * external * fixed in external * add changeset * close demo --------- Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>
26 lines
798 B
Python
26 lines
798 B
Python
import pytest
|
|
import transformers
|
|
|
|
import gradio as gr
|
|
|
|
|
|
@pytest.mark.flaky
|
|
def test_text_to_text_model_from_pipeline():
|
|
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)
|
|
|
|
|
|
@pytest.mark.flaky
|
|
def test_interface_in_blocks():
|
|
pipe1 = transformers.pipeline(model="sshleifer/bart-tiny-random")
|
|
pipe2 = transformers.pipeline(model="sshleifer/bart-tiny-random")
|
|
with gr.Blocks() as demo:
|
|
with gr.Tab("Image Inference"):
|
|
gr.Interface.from_pipeline(pipe1)
|
|
with gr.Tab("Image Inference"):
|
|
gr.Interface.from_pipeline(pipe2)
|
|
demo.launch(prevent_thread_lock=True)
|
|
demo.close()
|