token

test_gradio_client.py

@@ -1,10 +1,12 @@
-from gradio_client import Client
+from gradio_client import Client, file
 import requests
 import os
 
-client = Client("artificialguybr/Stable-Audio-Open-Zero")
+HF_TOKEN = os.getenv('HF_TOKEN')
+
+client = Client("dskill/sd-audio", hf_token=HF_TOKEN, download_files=True)
 result = client.predict(
-		prompt="Ambient rainfallz",
+		prompt="Ambient rainfall",
 		seconds_total=5,
 		steps=100,
 		cfg_scale=7,
@@ -12,6 +14,8 @@ result = client.predict(
 )
 print(result)
 
+
+
 # Split the path into parts
 parts = result.split(os.sep)
 
@@ -19,7 +23,7 @@ parts = result.split(os.sep)
 last_two_sections = os.sep.join(parts[-2:])
 print(last_two_sections)
 
-url_root = 'https://artificialguybr-stable-audio-open-zero.hf.space/file=/tmp/gradio/'
+url_root = 'https://dskill-sd-audio.hf.space/file=/tmp/gradio/'
 url = url_root + last_two_sections
 
 # Send a GET request to the URL