Upload app.py

app.py

@@ -18,9 +18,12 @@ 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 []
+    try:
+        new_content = content.decode(encoding='utf-8')
+        if new_content.startswith("error: "):
+            return []
+    except:
+        pass
 
     cache_dir = 'images'
     os.makedirs(cache_dir, exist_ok=True)