Synced repo using 'sync_with_huggingface' Github Action

app.py

@@ -130,17 +130,13 @@ async def img2img(pos_prompt: str, neg_promt: str = "", img = None,model="absolu
     loop = asyncio.get_running_loop()
     client = await loop.run_in_executor(None, get_client, None)
     img2img_args = img2img_conf.values()
-    job = client.submit(*img2img_args, fn_index=1)
-    
-    # img = Image.open(img)
-    # img.show()
-    await wait(job)
+    job = client.predict(*img2img_args, fn_index=1)
     return job.result()
 
-def run_dffusion(pos_prompt: str,neg_promt: str = "",img_url= None,model="absolutereality_v181.safetensors [3d9d4d2b]"):
+def run_dffusion(pos_prompt: str,neg_promt: str = "",img_url= "",model="absolutereality_v181.safetensors [3d9d4d2b]"):
     """Runs the diffusion model."""
 
-    if img_url == None:
+    if len(img_url) == 0:
         # Support for text prompts
         generated_image = text2img(pos_prompt, neg_promt,model)
     else:
@@ -151,8 +147,14 @@ def run_dffusion(pos_prompt: str,neg_promt: str = "",img_url= None,model="absolu
 
 
 # AI prediction command
-@bot.slash_command(name='diffusion',description='AI command to generate images',)
-async def diffusion(ctx, pos_prompt: str="",neg_promt: str = "", img_url= None,improve : Literal["True","False"] = "False",model : models = "absolutereality_v181.safetensors [3d9d4d2b]"):
+@bot.slash_command(name='diffusion',description='AI command to generate images')
+async def diffusion(ctx, 
+                    pos_prompt: discord.option(discord.SlashCommandOptionType.string, description="The positive prompt for the image generation."),
+                    neg_promt:  discord.option(discord.SlashCommandOptionType.string, description="The negative prompt for the image generation.", default=""),
+                    img_url= discord.option(discord.SlashCommandOptionType.string, description="The input image for the image generation.", default=""),
+                    improve : discord.option(discord.SlashCommandOptionType.string,choices = ["True","False"], description="Improve the results", default="False"),
+                    model : discord.option(discord.SlashCommandOptionType.string,choices = models, description="The model to use", default="absolutereality_v181.safetensors [3d9d4d2b]")
+                    ):
     """ 
     AI command to generate images
     """