v0.6 deepfloydif test

app.py

@@ -4,9 +4,15 @@ import gradio as gr
 import os
 import threading
 
+#todos
+#alert
+
 # Get Gradio client
 jojogan = gradio_client.Client("akhaliq/JoJoGAN")
 
+#new
+deepfloydif = gradio_client.Client("DeepFloyd/IF")
+
 # Set up discord bot 
 class MyClient(discord.Client):
     async def on_ready(self):
@@ -55,6 +61,19 @@ class MyClient(discord.Client):
                 await message.reply(f'Here is the {style} version of it', file=discord.File(im))
             else:
                 await message.channel.send("No attachments to be found...Can't animify dat! Try sending me an image 😉")
+
+        #new
+        #deepfloydif
+        if message.content.startswith('!deepfloydif'):
+            text_input = message.content[12:].strip()
+            if text_input:
+                im = deepfloydif.predict(text_input)
+                im_bytes = BytesIO()
+                im.save(im_bytes, 'PNG')
+                im_bytes.seek(0)
+                await message.reply(f'Here is the image generated for the input "{text_input}"', file=discord.File(im_bytes, 'output.png'))
+            else:
+                await message.channel.send("No text input provided. Please provide some text after the !deepfloydif command.")