Update app.py

app.py

@@ -1,8 +1,13 @@
+import os 
+os.system("pip install huggingface_hub")
+
+from huggingface_hub import space_info
+
 from predict import *
 from reconstructor import *
 from transformers import BertTokenizer, GPT2LMHeadModel
 
-import os
+#import os
 import gradio as gr
 
 model_path = "svjack/gpt-daliy-dialogue"
@@ -26,6 +31,15 @@ def demo_func(prefix, max_length):
         "Dialogue Context": l
     }
 
+markdown_exp_size = "##"
+lora_repo = "svjack/chatglm3-few-shot"
+lora_repo_link = "svjack/chatglm3-few-shot/?input_list_index=10"
+emoji_info = space_info(lora_repo).__dict__["cardData"]["emoji"]
+space_cnt = 1
+task_name = "[---Chinese Dialogue Generator---]"
+description = f"{markdown_exp_size} {task_name} few shot prompt in ChatGLM3 Few Shot space repo (click submit to activate) : [{lora_repo_link}](https://huggingface.co/spaces/{lora_repo_link}) {emoji_info}"
+
+
 demo = gr.Interface(
         fn=demo_func,
         inputs=[gr.Text(label = "Prefix"),
@@ -34,7 +48,8 @@ demo = gr.Interface(
         outputs="json",
         title=f"GPT Chinese Daliy Dialogue Generator 🐰 demonstration",
         examples=example_sample if example_sample else None,
-        description = 'This _example_ was **drive** from <br/><b><h4>[https://github.com/svjack/Daliy-Dialogue](https://github.com/svjack/Daliy-Dialogue)</h4></b>\n',
+        #description = 'This _example_ was **drive** from <br/><b><h4>[https://github.com/svjack/Daliy-Dialogue](https://github.com/svjack/Daliy-Dialogue)</h4></b>\n',
+        description = description,
         cache_examples = False
     )