Update chain_app.py

chain_app.py

@@ -11,7 +11,7 @@ openai_api_key = os.environ.get('OPENAI_API_KEY')
 groq_api_key = os.environ.get('GROQ_API_KEY')
 
 hf_text_client = Client("Artin2009/text-generation", hf_token=hf_token)
-hf_image_client = Client('Artin2009/image-generation')
+# hf_image_client = Client('Artin2009/image-generation')
 openai_client = OpenAI(api_key=openai_api_key)
 groq_client = Groq(api_key=groq_api_key)
 
@@ -38,10 +38,10 @@ async def chat_profile():
  name="Dorna-AI",
  markdown_description="One of the open-sourced models that neural brain team fine-tuned",
  ),
- cl.ChatProfile(
- name='Image-Generation',
- markdown_description='Our image generation model, has a performance like midjourney',
- ),
+ # cl.ChatProfile(
+ # name='Image-Generation',
+ # markdown_description='Our image generation model, has a performance like midjourney',
+ # ),
  cl.ChatProfile(
  name="GPT-4",
  markdown_description="OpenAI's GPT-4 model",
@@ -147,12 +147,12 @@ async def on_chat_start():
  content='my name is Dorna, Your AI Assistant designed by neural nexus team. i was made by Artin Daneshvar and Sadra Noadoust, 2 iranian students!'
  ).send()
 
- if chat_profile == 'Image-Generation':
- image = cl.Image(path='cat.png', name="result", display="inline")
- await cl.Message(
- content="I can make high quality & resoloution images for you, This is an example of what i can do!",
- elements=[image],
- ).send()
+ # if chat_profile == 'Image-Generation':
+ # image = cl.Image(path='cat.png', name="result", display="inline")
+ # await cl.Message(
+ # content="I can make high quality & resoloution images for you, This is an example of what i can do!",
+ # elements=[image],
+ # ).send()
  if chat_profile == 'GPT-4':
  await cl.ChatSettings(
  [
@@ -439,23 +439,23 @@ async def main(message: cl.Message):
  content=model_response
  ).send()
  
- elif chat_profile == 'Image-Generation':
- result = hf_image_client.predict(
- prompt=message.content,
- negative_prompt="",
- seed=0,
- randomize_seed=True,
- width=512,
- height=512,
- guidance_scale=0,
- num_inference_steps=2,
- api_name="/infer"
- )
- image = cl.Image(path=result, name="result", display="inline")
- await cl.Message(
- content="This message has an image!",
- elements=[image],
- ).send()
+ # elif chat_profile == 'Image-Generation':
+ # result = hf_image_client.predict(
+ # prompt=message.content,
+ # negative_prompt="",
+ # seed=0,
+ # randomize_seed=True,
+ # width=512,
+ # height=512,
+ # guidance_scale=0,
+ # num_inference_steps=2,
+ # api_name="/infer"
+ # )
+ # image = cl.Image(path=result, name="result", display="inline")
+ # await cl.Message(
+ # content="This message has an image!",
+ # elements=[image],
+ # ).send()
  elif chat_profile == 'GPT-4':
  completion = openai_client.chat.completions.create(
  model="gpt-4",