Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- .gitignore +3 -0
- fusion-app/app_api.py +27 -4
.gitignore
CHANGED
|
@@ -24,6 +24,8 @@ share/python-wheels/
|
|
| 24 |
*.egg-info/
|
| 25 |
.installed.cfg
|
| 26 |
*.egg
|
|
|
|
|
|
|
| 27 |
MANIFEST
|
| 28 |
|
| 29 |
# PyInstaller
|
|
@@ -206,3 +208,4 @@ marimo/_static/
|
|
| 206 |
marimo/_lsp/
|
| 207 |
__marimo__/
|
| 208 |
DSCS553_CS1_Assignment.pdf
|
|
|
|
|
|
| 24 |
*.egg-info/
|
| 25 |
.installed.cfg
|
| 26 |
*.egg
|
| 27 |
+
keys.txt
|
| 28 |
+
|
| 29 |
MANIFEST
|
| 30 |
|
| 31 |
# PyInstaller
|
|
|
|
| 208 |
marimo/_lsp/
|
| 209 |
__marimo__/
|
| 210 |
DSCS553_CS1_Assignment.pdf
|
| 211 |
+
key.txt
|
fusion-app/app_api.py
CHANGED
|
@@ -19,10 +19,33 @@ CSV_API = HERE / "runs_api.csv"
|
|
| 19 |
CLIP_MODEL = "openai/clip-vit-base-patch32"
|
| 20 |
W2V2_MODEL = "facebook/wav2vec2-base"
|
| 21 |
|
| 22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
if not HF_TOKEN:
|
| 24 |
-
print("Warning: HuggingFace token not found
|
| 25 |
-
print("To use the API version, set one of these environment variables: HF_Token, HF_TOKEN, or HUGGINGFACE_TOKEN")
|
| 26 |
client = None
|
| 27 |
else:
|
| 28 |
client = InferenceClient(token=HF_TOKEN)
|
|
@@ -64,7 +87,7 @@ def _wave_float32_to_wav_bytes(wave_16k: np.ndarray, sr=16000) -> bytes:
|
|
| 64 |
|
| 65 |
def w2v2_api_embed(wave_16k: np.ndarray) -> np.ndarray:
|
| 66 |
if HF_TOKEN is None:
|
| 67 |
-
raise RuntimeError("HuggingFace token not available.
|
| 68 |
|
| 69 |
wav_bytes = _wave_float32_to_wav_bytes(wave_16k)
|
| 70 |
|
|
|
|
| 19 |
CLIP_MODEL = "openai/clip-vit-base-patch32"
|
| 20 |
W2V2_MODEL = "facebook/wav2vec2-base"
|
| 21 |
|
| 22 |
+
def get_hf_token():
|
| 23 |
+
token = os.getenv("HF_TOKEN")
|
| 24 |
+
if token:
|
| 25 |
+
return token
|
| 26 |
+
|
| 27 |
+
key_paths = [
|
| 28 |
+
HERE / "key.txt",
|
| 29 |
+
HERE.parent / "key.txt",
|
| 30 |
+
Path("key.txt"),
|
| 31 |
+
Path("../key.txt")
|
| 32 |
+
]
|
| 33 |
+
|
| 34 |
+
for key_path in key_paths:
|
| 35 |
+
try:
|
| 36 |
+
if key_path.exists():
|
| 37 |
+
token = key_path.read_text().strip()
|
| 38 |
+
if token:
|
| 39 |
+
print(f"HuggingFace token loaded from {key_path}")
|
| 40 |
+
return token
|
| 41 |
+
except Exception as e:
|
| 42 |
+
print(f"Error reading {key_path}: {e}")
|
| 43 |
+
|
| 44 |
+
return None
|
| 45 |
+
|
| 46 |
+
HF_TOKEN = get_hf_token()
|
| 47 |
if not HF_TOKEN:
|
| 48 |
+
print("Warning: HuggingFace token not found. API functions will not work.")
|
|
|
|
| 49 |
client = None
|
| 50 |
else:
|
| 51 |
client = InferenceClient(token=HF_TOKEN)
|
|
|
|
| 87 |
|
| 88 |
def w2v2_api_embed(wave_16k: np.ndarray) -> np.ndarray:
|
| 89 |
if HF_TOKEN is None:
|
| 90 |
+
raise RuntimeError("HuggingFace token not available.")
|
| 91 |
|
| 92 |
wav_bytes = _wave_float32_to_wav_bytes(wave_16k)
|
| 93 |
|