Update models.py

models.py

@@ -35,7 +35,7 @@ class LlmBot():
  )
  system_setting = self.character_prompt({"name": name, "description": description, "user_name": user_name}, max_new_tokens)
  formatted_prompt = self.format_prompt(prompt, history, system_setting)
- stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
+ stream = self.client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
  
  return stream