Update app.py

app.py

@@ -1,14 +1,24 @@
 import torch
 import gradio as gr
 from transformers import pipeline
+from huggingface_hub import login
+import os
 import numpy as np
 
+# -----------------------------
+# HUGGING FACE LOGIN
+# -----------------------------
+HF_TOKEN = os.getenv("HF_TOKEN")
+if HF_TOKEN is None:
+    raise ValueError("Environment variable HF_TOKEN not found. Please set it before running.")
+
+login(HF_TOKEN)
+
 # -----------------------------
 # CONFIG
 # -----------------------------
 MODEL_NAME = "canopylabs/orpheus-3b-0.1-ft"
 DEVICE = 0 if torch.cuda.is_available() else -1
-SAMPLE_RATE = 24000
 
 # -----------------------------
 # LOAD PIPELINE
@@ -17,6 +27,7 @@ tts_pipe = pipeline(
     task="text-to-speech",
     model=MODEL_NAME,
     device=DEVICE,
+    use_auth_token=HF_TOKEN  # دسترسی به مدل gated
 )
 
 # -----------------------------
@@ -28,12 +39,9 @@ def tts_generate(text):
 
     output = tts_pipe(text)
 
-    audio = output["audio"]
+    audio = np.asarray(output["audio"], dtype=np.float32)
     sr = output["sampling_rate"]
 
-    # Ensure numpy float32
-    audio = np.asarray(audio, dtype=np.float32)
-
     return (sr, audio)
 
 # -----------------------------