File size: 779 Bytes
a03b3ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import pytest
import transformers
import gradio as gr
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()
|