Update custom_llm.py

custom_llm.py

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