Synced repo using 'sync_with_huggingface' Github Action
Browse files- app.py +1 -0
- deepfloydif.py +5 -1
- falcon.py +5 -1
- musicgen.py +5 -1
app.py
CHANGED
|
@@ -9,6 +9,7 @@ from discord.ext import commands
|
|
| 9 |
from falcon import continue_falcon, try_falcon
|
| 10 |
from musicgen import music_create
|
| 11 |
|
|
|
|
| 12 |
# HF GUILD SETTINGS
|
| 13 |
MY_GUILD_ID = 1077674588122648679 if os.getenv("TEST_ENV", False) else 879548962464493619
|
| 14 |
MY_GUILD = discord.Object(id=MY_GUILD_ID)
|
|
|
|
| 9 |
from falcon import continue_falcon, try_falcon
|
| 10 |
from musicgen import music_create
|
| 11 |
|
| 12 |
+
|
| 13 |
# HF GUILD SETTINGS
|
| 14 |
MY_GUILD_ID = 1077674588122648679 if os.getenv("TEST_ENV", False) else 879548962464493619
|
| 15 |
MY_GUILD = discord.Object(id=MY_GUILD_ID)
|
deepfloydif.py
CHANGED
|
@@ -97,7 +97,11 @@ async def deepfloydif_stage_1(ctx, prompt, client):
|
|
| 97 |
print("Safety checks passed for deepfloydif_stage_1")
|
| 98 |
# interaction.response message can't be used to create a thread, so we create another message
|
| 99 |
message = await ctx.send(f"**{prompt}** - {ctx.author.mention}")
|
| 100 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 101 |
await thread.send(
|
| 102 |
"[DISCLAIMER: HuggingBot is a **highly experimental** beta feature; Additional information on the"
|
| 103 |
" DeepfloydIF model can be found here: https://huggingface.co/spaces/DeepFloyd/IF"
|
|
|
|
| 97 |
print("Safety checks passed for deepfloydif_stage_1")
|
| 98 |
# interaction.response message can't be used to create a thread, so we create another message
|
| 99 |
message = await ctx.send(f"**{prompt}** - {ctx.author.mention}")
|
| 100 |
+
if len(prompt) > 99:
|
| 101 |
+
small_prompt = prompt[:99]
|
| 102 |
+
else:
|
| 103 |
+
small_prompt = prompt
|
| 104 |
+
thread = await message.create_thread(name=f"{small_prompt}", auto_archive_duration=60)
|
| 105 |
await thread.send(
|
| 106 |
"[DISCLAIMER: HuggingBot is a **highly experimental** beta feature; Additional information on the"
|
| 107 |
" DeepfloydIF model can be found here: https://huggingface.co/spaces/DeepFloyd/IF"
|
falcon.py
CHANGED
|
@@ -68,7 +68,11 @@ async def try_falcon(ctx, prompt):
|
|
| 68 |
if os.environ.get("TEST_ENV") == "True":
|
| 69 |
print("Safetychecks passed for try_falcon")
|
| 70 |
message = await ctx.send(f"**{prompt}** - {ctx.author.mention}")
|
| 71 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 72 |
await thread.send(
|
| 73 |
"[DISCLAIMER: HuggingBot is a **highly experimental** beta feature; The Falcon model and system"
|
| 74 |
" prompt can be found here: https://huggingface.co/spaces/HuggingFaceH4/falcon-chat]"
|
|
|
|
| 68 |
if os.environ.get("TEST_ENV") == "True":
|
| 69 |
print("Safetychecks passed for try_falcon")
|
| 70 |
message = await ctx.send(f"**{prompt}** - {ctx.author.mention}")
|
| 71 |
+
if len(prompt) > 99:
|
| 72 |
+
small_prompt = prompt[:99]
|
| 73 |
+
else:
|
| 74 |
+
small_prompt = prompt
|
| 75 |
+
thread = await message.create_thread(name=small_prompt, auto_archive_duration=60) # interaction.user
|
| 76 |
await thread.send(
|
| 77 |
"[DISCLAIMER: HuggingBot is a **highly experimental** beta feature; The Falcon model and system"
|
| 78 |
" prompt can be found here: https://huggingface.co/spaces/HuggingFaceH4/falcon-chat]"
|
musicgen.py
CHANGED
|
@@ -34,7 +34,11 @@ async def music_create(ctx, prompt):
|
|
| 34 |
print("Safetychecks passed for music_create")
|
| 35 |
|
| 36 |
message = await ctx.send(f"**{prompt}** - {ctx.author.mention}")
|
| 37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
|
| 39 |
await thread.send(
|
| 40 |
"[DISCLAIMER: HuggingBot is a beta feature; The MusicGen"
|
|
|
|
| 34 |
print("Safetychecks passed for music_create")
|
| 35 |
|
| 36 |
message = await ctx.send(f"**{prompt}** - {ctx.author.mention}")
|
| 37 |
+
if len(prompt) > 99:
|
| 38 |
+
small_prompt = prompt[:99]
|
| 39 |
+
else:
|
| 40 |
+
small_prompt = prompt
|
| 41 |
+
thread = await message.create_thread(name=small_prompt, auto_archive_duration=60)
|
| 42 |
|
| 43 |
await thread.send(
|
| 44 |
"[DISCLAIMER: HuggingBot is a beta feature; The MusicGen"
|