reach-vb HF staff commited on
Commit
9922ff1
1 Parent(s): 1a2d6a6
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,13 +27,13 @@ vibes = pipeline("text-to-audio",
27
  torch_dtype=torch.float16,
28
  device="cuda")
29
 
30
- client = InferenceClient(model="HuggingFaceH4/zephyr-7b-beta",)
31
 
32
 
33
  # Inference
34
  def generate_audio(text,):
35
  prompt = f"Take the next sentence and enrich it with details. Keep it compact. {text}"
36
- output = client.text_generation(prompt, max_new_tokens=100)
37
  out = vibes(output)
38
 
39
  with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as f:
 
27
  torch_dtype=torch.float16,
28
  device="cuda")
29
 
30
+ client = InferenceClient(model="mistralai/Mixtral-8x7B-Instruct-v0.1",)
31
 
32
 
33
  # Inference
34
  def generate_audio(text,):
35
  prompt = f"Take the next sentence and enrich it with details. Keep it compact. {text}"
36
+ output = client.text_generation(prompt, max_new_tokens=250)
37
  out = vibes(output)
38
 
39
  with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as f: