update

app.py

@@ -149,19 +149,24 @@ def inference_gen(text, task, do_sample, top_p, top_k, max_token, temperature, b
     #print(generate_config)
     #print(text)
     default_example = default_example and generate_config == default_generate_config
-    try:
-        if default_example:
-            prediction = cl.generate(
-                model_name='clueai-base',
-                prompt=text)
-        else:
-            prediction = cl.generate(
-                model_name='clueai-base',
-                prompt=text,
-                generate_config=generate_config)
-    except Exception as e:
-        logger.error(f"error, {e}")
-        return
+    try_num = 3
+    while try_num:
+        try:
+            if default_example:
+                prediction = cl.generate(
+                    model_name='clueai-base',
+                    prompt=text)
+            else:
+                prediction = cl.generate(
+                    model_name='clueai-base',
+                    prompt=text,
+                    generate_config=generate_config)
+        except Exception as e:
+            logger.error(f"error, {e}")
+            return
+        if prediction.generations[0].text != "含有违规词，不予展示":
+            break
+        try_num -= 1
 
     return prediction.generations[0].text