Upload app.py

app.py

@@ -18,6 +18,9 @@ def gen_image(prompt: str):
     url = f"https://aip.baidubce.com/rpc/2.0/nlp-itec/poc/chinese_stable_diffusion?access_token={access_token}"
 
     content = requests.post(url, json={"text": prompt}).content
+    new_content = content.decode(encoding='utf-8')
+    if new_content.startswith("error: "):
+        return []
 
     cache_dir = 'images'
     os.makedirs(cache_dir, exist_ok=True)
@@ -46,8 +49,10 @@ examples = [
     '厚涂层油画《悲伤男孩的特写肖像》，本·科迪，希卡里·希莫达'
 ]
 
+
 with block:
     gr.HTML(read_content("header.html"))
+    gr.Markdown("> warning", elem_id='warning', visible=False)
     gr.Markdown("[![Stargazers repo roster for @PaddlePaddle/PaddleNLP](https://reporoster.com/stars/PaddlePaddle/PaddleNLP)](https://github.com/PaddlePaddle/PaddleNLP)")
     with gr.Group():
         with gr.Box():