Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 22 |
print("Loading SNAC model...")
|
| 23 |
snac = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").to(device)
|
| 24 |
|
| 25 |
-
model_name = "
|
| 26 |
# optional: explizites snapshot_download (entfernt große Dateien)
|
| 27 |
|
| 28 |
snapshot_download(
|
|
@@ -108,7 +108,7 @@ async def tts_ws(ws: WebSocket):
|
|
| 108 |
msg = await ws.receive_text()
|
| 109 |
data = json.loads(msg)
|
| 110 |
text = data.get("text", "")
|
| 111 |
-
voice = data.get("voice", "
|
| 112 |
|
| 113 |
# 1) Prompt → Tokens
|
| 114 |
ids, mask = process_prompt(text, voice)
|
|
|
|
| 22 |
print("Loading SNAC model...")
|
| 23 |
snac = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").to(device)
|
| 24 |
|
| 25 |
+
model_name = "SebastianBodza/Kartoffel_Orpheus-3B_german_natural-v0.1"
|
| 26 |
# optional: explizites snapshot_download (entfernt große Dateien)
|
| 27 |
|
| 28 |
snapshot_download(
|
|
|
|
| 108 |
msg = await ws.receive_text()
|
| 109 |
data = json.loads(msg)
|
| 110 |
text = data.get("text", "")
|
| 111 |
+
voice = data.get("voice", "Jakob")
|
| 112 |
|
| 113 |
# 1) Prompt → Tokens
|
| 114 |
ids, mask = process_prompt(text, voice)
|