SonicaB commited on
Commit
138d478
·
verified ·
1 Parent(s): b225c8e

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. .gitignore +3 -0
  2. 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
- HF_TOKEN = os.getenv("HF_Token") or os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
  if not HF_TOKEN:
24
- print("Warning: HuggingFace token not found in environment. API functions will not work.")
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. Please set HF_Token environment variable.")
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