guyyariv commited on
Commit
411af0b
1 Parent(s): 62de77b

AudioTokenDemo

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -114,7 +114,7 @@ def greet(audio):
114
 
115
  if __name__ == "__main__":
116
 
117
- lora = True
118
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
119
  model = AudioTokenWrapper(lora, device)
120
 
@@ -125,9 +125,9 @@ if __name__ == "__main__":
125
  </p>"""
126
 
127
  examples = [
128
- ["assets/train.wav"],
129
  ["assets/dog barking.wav"],
130
- # ["assets/airplane.wav"],
131
  # ["assets/electric guitar.wav"],
132
  # ["assets/female sings.wav"],
133
  ]
 
114
 
115
  if __name__ == "__main__":
116
 
117
+ lora = False
118
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
119
  model = AudioTokenWrapper(lora, device)
120
 
 
125
  </p>"""
126
 
127
  examples = [
128
+ # ["assets/train.wav"],
129
  ["assets/dog barking.wav"],
130
+ ["assets/airplane.wav"],
131
  # ["assets/electric guitar.wav"],
132
  # ["assets/female sings.wav"],
133
  ]