fix path

bghira · May 28, 2024 · 05915ab · 05915ab
1 parent f571ca6
commit 05915ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/discord_tron_master/cogs/image/__init__.py b/discord_tron_master/cogs/image/__init__.py
@@ -32,8 +32,8 @@ def generate_terminus_via_hub(prompt: str, model: str = "velocity", user_id: int
             negative_prompt=user_config.get("negative_prompt", "underexposed, blurry, ugly, washed-out"),
             api_name="/predict"
     )
-    split_pieces = result.split('/')
-    return f"https://ptx0-ptx0-terminus-xl-velocity-v2.hf.space/file=/tmp/gradio/{split_pieces[-2]}/image.png"
+    split_pieces = result[0]['image'].split('/')
+    return f"https://ptx0-ptx0-terminus-xl-velocity-v2.hf.space/file=/tmp/gradio/{split_pieces[-2]}/image.webp"
 
 def generate_lumina_image(prompt: str, use_5b: bool = False):
     from gradio_client import Client