handle invalid output from models

eosphoros-ai · Dec 1, 2023 · b35533e · b35533e
1 parent 8359ba8
commit b35533e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dbgpt_hub/predict/predict.py b/dbgpt_hub/predict/predict.py
@@ -49,7 +49,10 @@ def main():
 
     with open(predict_output_dir_name, "w") as f:
         for p in result:
-            f.write(p.replace("\n", " ") + "\n")
+            try:
+                f.write(p.replace("\n", " ") + "\n")
+            except:
+                f.write("Invalid Output!\n")
 
 
 if __name__ == "__main__":