Update custom_llm.py

custom_llm.py

@@ -174,8 +174,8 @@ def out_format(text, llm, df):
     for i in range(6):
 
         try :
-            # text = text.split("```python")[-1].split("```")[0]
-            text = text.split("# Generated Code")[-1].split("`")[0]
+            text = text.split("```python")[-1].split("```")[0]
+            # text = text.split("# Generated Code")[-1].split("```")[0]
             print(text)
             exec(text)