Update custom_llm.py

custom_llm.py

@@ -182,7 +182,7 @@ def out_format(text, llm, df):
             text_split = text.split("```python")[-1].split("```")[0].replace('\_', "_")
             # text = text.split("# Generated Code")[-1].split("```")[0]
             if "response" not in text_split:
-                text = text.split("```")[0]
+                text = text.split("```")[0].replace('\_', "_")
             else :
                 text = text_split