Update custom_llm.py

custom_llm.py

@@ -145,7 +145,7 @@ def format_df(df):
     return out
 
 
-def out_format(text, df):
+def out_format(text, llm, df):
 
     prompt = PromptTemplate.from_template("""<s><INST>Fix the following code:
     {code}
@@ -209,7 +209,7 @@ def custom_dataframe_chain(llm, df):
     
     </INST></s>""")
 
-    return prompt | llm | RunnableLambda(lambda x:out_format(x, df))
+    return prompt | llm | RunnableLambda(lambda x:out_format(x, llm, df))