not-lain commited on
Commit
560c226
·
1 Parent(s): c410fef

Synced repo using 'sync_with_huggingface' Github Action

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -60,7 +60,7 @@ async def on_ready():
60
  name="ping",
61
  description="ping")
62
  async def ping(ctx):
63
- await ctx.respond(f"{bot.latency*1000:.0f}ms")
64
 
65
 
66
  def get_client(session: Optional[str] = None) -> grc.Client:
@@ -99,7 +99,7 @@ async def text2img(pos_prompt: str, neg_promt: str = "",model="absolutereality_v
99
  txt2img_args = txt2img_conf.values()
100
  job = client.predict(*txt2img_args, fn_index=0)
101
 
102
- return job.result()
103
 
104
 
105
  async def img2img(pos_prompt: str, neg_promt: str = "", img = None,model="absolutereality_v181.safetensors [3d9d4d2b]"):
@@ -133,7 +133,7 @@ async def img2img(pos_prompt: str, neg_promt: str = "", img = None,model="absolu
133
  client = await loop.run_in_executor(None, get_client, None)
134
  img2img_args = img2img_conf.values()
135
  job = client.predict(*img2img_args, fn_index=1)
136
- return job.result()
137
 
138
  def run_dffusion(pos_prompt: str,neg_promt: str = "",img_url= "",model="absolutereality_v181.safetensors [3d9d4d2b]"):
139
  """Runs the diffusion model."""
@@ -160,7 +160,7 @@ async def diffusion(ctx, pos_prompt: str="",neg_promt: str = "", img_url= None,i
160
  # if channel name is == channel_name
161
  try:
162
  if ctx.channel.name == channel_name:
163
- await ctx.respond(f"Creating image for {ctx.author.mention} ...")
164
  try :
165
  # preparing the prediction before creating the thread
166
  _improved_pos_prompt = "ultrarealistic,8k"
@@ -184,7 +184,7 @@ async def diffusion(ctx, pos_prompt: str="",neg_promt: str = "", img_url= None,i
184
 
185
  # need to make sure AI sends the first message
186
  result = run_dffusion(pos_prompt , neg_promt,img_url,model=model)
187
- await ctx.resepond(result)
188
  except Exception as e:
189
  await ctx.resepond(e)
190
  else:
@@ -203,15 +203,15 @@ async def diffusion(ctx, pos_prompt: str="",neg_promt: str = "", img_url= None,i
203
  description="setup the bot")
204
  async def setup(ctx):
205
  """
206
- create the #falcon-180b-demo channel
207
  """
208
  # if channel falcon-180b-demo doesn't exist create it
209
  if not discord.utils.get(ctx.guild.channels, name=channel_name):
210
  await ctx.guild.create_text_channel(channel_name,category=ctx.channel.category)
211
- await ctx.respond(channel_name)
212
  else:
213
  # TODO: tag the channel
214
- await ctx.respond("#falcon-180b-demo channel already exist")
215
 
216
  # running in thread
217
  def run_bot():
 
60
  name="ping",
61
  description="ping")
62
  async def ping(ctx):
63
+ await ctx.reply(f"{bot.latency*1000:.0f}ms")
64
 
65
 
66
  def get_client(session: Optional[str] = None) -> grc.Client:
 
99
  txt2img_args = txt2img_conf.values()
100
  job = client.predict(*txt2img_args, fn_index=0)
101
 
102
+ return job
103
 
104
 
105
  async def img2img(pos_prompt: str, neg_promt: str = "", img = None,model="absolutereality_v181.safetensors [3d9d4d2b]"):
 
133
  client = await loop.run_in_executor(None, get_client, None)
134
  img2img_args = img2img_conf.values()
135
  job = client.predict(*img2img_args, fn_index=1)
136
+ return job
137
 
138
  def run_dffusion(pos_prompt: str,neg_promt: str = "",img_url= "",model="absolutereality_v181.safetensors [3d9d4d2b]"):
139
  """Runs the diffusion model."""
 
160
  # if channel name is == channel_name
161
  try:
162
  if ctx.channel.name == channel_name:
163
+ await ctx.reply(f"Creating image for {ctx.author.mention} ...")
164
  try :
165
  # preparing the prediction before creating the thread
166
  _improved_pos_prompt = "ultrarealistic,8k"
 
184
 
185
  # need to make sure AI sends the first message
186
  result = run_dffusion(pos_prompt , neg_promt,img_url,model=model)
187
+ await ctx.channel.send(file=discord.File(result))
188
  except Exception as e:
189
  await ctx.resepond(e)
190
  else:
 
203
  description="setup the bot")
204
  async def setup(ctx):
205
  """
206
+ create the #channel_name channel
207
  """
208
  # if channel falcon-180b-demo doesn't exist create it
209
  if not discord.utils.get(ctx.guild.channels, name=channel_name):
210
  await ctx.guild.create_text_channel(channel_name,category=ctx.channel.category)
211
+ await ctx.reply(channel_name)
212
  else:
213
  # TODO: tag the channel
214
+ await ctx.reply(f"{channel_name} channel already exist")
215
 
216
  # running in thread
217
  def run_bot():