Update custom_llm.py

custom_llm.py

@@ -17,7 +17,7 @@ class CustomLLM(LLM):
         if stop is not None:
             raise ValueError("stop kwargs are not permitted.")
 
-        response = co.chat(message=f'{prompt}', max_tokens=1700, temperature=0.9)
+        response = co.chat(model = self.model, message=f'{prompt}', max_tokens=1700, temperature=0.9)
         print(response.text)
 
         return f' ```python \n {response.text} \n ``` '