Spaces:
Running
on
T4
Running
on
T4
Update app.py (#2)
Browse files- Update app.py (e360f5c793401c5ce465d8a836c0015c3dfed11d)
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="
|
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=
|
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:
|