Spaces:
Runtime error
Runtime error
File size: 525 Bytes
9f6a8e3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import numpy as np
import gradio as gr
import paddlehub as hub
model = hub.Module(name='ernie_vilg')
def inference(text_prompts):
results = model.generate_image(
text_prompts=text_prompts)
return np.array(results[0])
outputs = [
gr.outputs.Image(type="numpy",label="result")
]
title="ERNIE-ViLG"
description="ERNIE-ViLG model, which supports text-to-image task."
gr.Interface(inference, gr.inputs.Textbox(),outputs,title=title,description=description).launch(enable_queue=True) |