test

app.py

@@ -1,8 +1,8 @@
-import matplotlib.pyplot as plt
+# import matplotlib.pyplot as plt
 import gradio as gr
-import paddlehub as hub
+# import paddlehub as hub
 
-ernie_zeus = hub.Module(name='ernie_zeus')
+# ernie_zeus = hub.Module(name='ernie_zeus')
 
 
 def inference(task: str,
@@ -14,9 +14,9 @@ def inference(task: str,
     return "success"
 
 
-title = "ERNIE-ViLG"
+title = "ERNIE-Zeus"
 
-description = "ERNIE-ViLG model, which supports text-to-image task."
+description = "ERNIE-Zeus model, which supports Chinese text generates task."
 
 css = """
         .gradio-container {
@@ -134,12 +134,11 @@ with block:
                   justify-content: center;
                 ">
               <h1 style="font-weight: 900; margin-bottom: 7px;">
-                  ERNIE-ViLG Demo
+                  ERNIE-Zeus Demo
               </h1>
               </div> 
               <p style="margin-bottom: 10px; font-size: 94%">
-                ERNIE-ViLG is a state-of-the-art text-to-image model that generates
-                images from Chinese text.
+                ERNIE-Zeus is a state-of-the-art Chinese text generates model.
               </p>
             </div>
         """
@@ -170,9 +169,9 @@ with block:
         topp = gr.Number(label="topp", value=1.0)
         penalty_score = gr.Number(label="penalty_score", value=1.0)
         
-        text_gen = gr.Gallery(
-            value=['images/%02d.jpg' % i for i in range(1, 7)], label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[2, 3], height="auto")
+        text_gen = gr.Text(
+            value='哈哈哈哈哈', label="Generated images", show_label=False
+        )
 
         ex = gr.Examples(examples=examples, fn=inference, inputs=[
                          text, task], outputs=text_gen, cache_examples=False)
@@ -182,12 +181,6 @@ with block:
         btn.click(inference, inputs=[text, task], outputs=text_gen)
         gr.HTML(
             """
-                <div class="prompt">
-                    <p><h4>Prompt公式</h4>
-                    <span> Prompt = [形容词] [主语] ，[细节设定]， [修饰语或者艺术家]。 </span>
-                    关于各部分的构造方式和效果，可以参考<a href="https://github.com/PaddlePaddle/PaddleHub/tree/develop/modules/image/text_to_image/ernie_vilg#六-prompt-指南" style="text-decoration: underline;" target="_blank">YouPromptMe指南</a>。
-                    </p>   
-               </div>
                 <div class="footer">
                     <p>Model by <a href="https://github.com/PaddlePaddle/PaddleHub" style="text-decoration: underline;" target="_blank">PaddleHub</a> and <a href="https://wenxin.baidu.com" style="text-decoration: underline;" target="_blank">文心大模型</a> - Gradio Demo by 🤗 Hugging Face
                     </p>