lunarflu HF Staff commited on
Commit
425245c
·
1 Parent(s): 00b58e3

added [test]

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -72,6 +72,18 @@ async def deepfloydif(interaction: discord.Interaction, prompt: str):
72
  print(f"Error: {e}")
73
 
74
 
 
 
 
 
 
 
 
 
 
 
 
 
75
  @client.event
76
  async def on_reaction_add(reaction, user):
77
  """Checks for a reaction in order to call dfif2"""
 
72
  print(f"Error: {e}")
73
 
74
 
75
+ @client.tree.command()
76
+ @app_commands.describe(
77
+ prompt="Enter a prompt to generate an image! Can generate realistic text, too!"
78
+ )
79
+ async def test(interaction: discord.Interaction, prompt: str):
80
+ """DeepfloydIF stage 1 generation"""
81
+ try:
82
+ await interaction.response.send_message(f"{prompt}")
83
+ except Exception as e:
84
+ print(f"Error: {e}")
85
+
86
+
87
  @client.event
88
  async def on_reaction_add(reaction, user):
89
  """Checks for a reaction in order to call dfif2"""