test

app.py

@@ -10,9 +10,13 @@ def inference(task: str,
               seq_len: int = 512,
               topp: float = 0.9,
               penalty_score: float = 1.0):
-    func = getattr(ernie_zeus, task)
-    return func(text, min_dec_len, seq_len, topp, penalty_score)
 
+    func = getattr(ernie_zeus, task)
+    try:
+        result = func(text, min_dec_len, seq_len, topp, penalty_score)
+        return result
+    except Exception as error:
+        return str(error)
 
 title = "ERNIE-Zeus"
 
@@ -197,7 +201,7 @@ with block:
         min_dec_len = gr.Slider(minimum=1, maximum=511, value=1, label="min_dec_len", step=1, interactive=True)
         seq_len = gr.Slider(minimum=2, maximum=512, value=128, label="seq_len", step=1, interactive=True)
         topp = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, label="topp", step=0.01, interactive=True)
-        penalty_score = gr.Slider(minimum=1.0, maximum=2.0, value=1.0, label="topp", step=0.01, interactive=True)
+        penalty_score = gr.Slider(minimum=1.0, maximum=2.0, value=1.0, label="penalty_score", step=0.01, interactive=True)
 
         text_gen = gr.Text(label="generated_text")