add ERNIE-ViLG module

README.md

@@ -1,5 +1,5 @@
 ---
-title: ERNIE ViLG
+title: ERNIE-ViLG
 emoji: 📉
 colorFrom: yellow
 colorTo: pink

app.py

@@ -0,0 +1,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)

requirements.txt

@@ -0,0 +1,2 @@
+paddlehub
+requests