theOnlyJaco commited on
Commit
971e667
·
unverified ·
1 Parent(s): 148d2c2

Change to demo

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -18,6 +18,7 @@ class ClapSSGradio():
18
  def __init__(
19
  self,
20
  name,
 
21
  k=10,
22
  ):
23
 
@@ -25,9 +26,9 @@ class ClapSSGradio():
25
  self.k = k
26
 
27
  self.model = ClapModel.from_pretrained(
28
- f"Audiogen/{name}", use_auth_token=os.getenv('HUGGINGFACE_API_TOKEN'))
29
  self.processor = ClapProcessor.from_pretrained(
30
- f"Audiogen/{name}", use_auth_token=os.getenv('HUGGINGFACE_API_TOKEN'))
31
 
32
  self.sas_token = os.environ['AZURE_SAS_TOKEN']
33
  self.account_name = 'Audiogen'
@@ -138,5 +139,5 @@ class ClapSSGradio():
138
 
139
 
140
  if __name__ == "__main__":
141
- app = ClapSSGradio("clap-2")
142
  app.launch(share=False)
 
18
  def __init__(
19
  self,
20
  name,
21
+ model = "clap-2",
22
  k=10,
23
  ):
24
 
 
26
  self.k = k
27
 
28
  self.model = ClapModel.from_pretrained(
29
+ f"Audiogen/{model}", use_auth_token=os.getenv('HUGGINGFACE_API_TOKEN'))
30
  self.processor = ClapProcessor.from_pretrained(
31
+ f"Audiogen/{model}", use_auth_token=os.getenv('HUGGINGFACE_API_TOKEN'))
32
 
33
  self.sas_token = os.environ['AZURE_SAS_TOKEN']
34
  self.account_name = 'Audiogen'
 
139
 
140
 
141
  if __name__ == "__main__":
142
+ app = ClapSSGradio("demo")
143
  app.launch(share=False)