Update app.py

app.py

@@ -9,7 +9,7 @@ import discord
 from discord.ext import tasks
 from PIL import Image
 import streamlit as st
-from huggingface_hub import InferenceClient, login
+from huggingface_hub import AsyncInferenceClient, login
 import time
 import os
 
@@ -71,9 +71,9 @@ HF_TOKEN = os.environ["HF_TOKEN"]
 
 login(HF_TOKEN)
 
-SD = InferenceClient(model="dataautogpt3/ProteusV0.2")
-LLM = InferenceClient(model="openchat/openchat-3.5-0106")
-RF = InferenceClient(model="stabilityai/stable-diffusion-xl-refiner-1.0")
+SD = AsyncInferenceClient(model="dataautogpt3/ProteusV0.2")
+LLM = AsyncInferenceClient(model="openchat/openchat-3.5-0106")
+RF = AsyncInferenceClient(model="stabilityai/stable-diffusion-xl-refiner-1.0")
 
 
 def ec(x, fd="<|image|>", sd="<|image|>"):
@@ -205,7 +205,7 @@ async def on_message(message):
                 e = await message.reply(embed=embed)
             if imgp != "":
                 try:
-                    image = SD.text_to_image(imgp)
+                    image = await SD.text_to_image(imgp)
                     image.save("image.png")
                     embed.set_footer(
                         text=
@@ -214,7 +214,7 @@ async def on_message(message):
                     file = discord.File("image.png", filename="image.png")
                     embed.set_image(url="attachment://image.png")
                     await e.edit(embed=embed, attachments=[file])
-                    image = RF.image_to_image("image.png")
+                    image = await RF.image_to_image("image.png")
                     embed.set_footer(
                         text=
                         """Information or code generated by Lyre may not always be correct. Lyre was made by Araeyn."""