Drew commited on
Commit
e437aae
1 Parent(s): a8c3afa

switch API

Browse files
Files changed (1) hide show
  1. test_gradio_client.py +3 -2
test_gradio_client.py CHANGED
@@ -6,11 +6,12 @@ HF_TOKEN = os.getenv('HF_TOKEN')
6
 
7
  client = Client("dskill/sd-audio", hf_token=HF_TOKEN, download_files=True)
8
  result = client.predict(
9
- prompt="Ambient rainfall",
10
  seconds_total=5,
11
  steps=100,
12
  cfg_scale=7,
13
- api_name="/predict"
14
  )
 
15
  #this is on your local machine wtf wild
16
  print("result: " + result)
 
6
 
7
  client = Client("dskill/sd-audio", hf_token=HF_TOKEN, download_files=True)
8
  result = client.predict(
9
+ prompt="a fart",
10
  seconds_total=5,
11
  steps=100,
12
  cfg_scale=7,
13
+ api_name="/generate_audio"
14
  )
15
+
16
  #this is on your local machine wtf wild
17
  print("result: " + result)