Somalitts commited on
Commit
0114c33
·
verified ·
1 Parent(s): 8308695

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -11
app.py CHANGED
@@ -20,19 +20,22 @@ app.add_middleware(
20
  # --- Soo Dejinta Moodeelka (Model Loading) ---
21
  # Kani hadda waxa uu isticmaali doonaa jidka keydka (cache path) ee lagu dejiyay Dockerfile-ka
22
  # oo ah /app/hf-cache, kaas oo leh ruqsadaha saxda ah.
 
 
 
 
 
 
23
  MODEL_ID = "Mustafaa4a/ASR-Somali"
24
- processor = None
25
- model = None
26
-
27
- print("Soo dejinta moodeelka iyo processor-ka...")
28
- try:
29
- processor = Wav2Vec2Processor.from_pretrained(MODEL_ID)
30
- model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
31
- print("Moodeelka iyo processor-ka si guul leh ayaa loo soo dejiyay.")
32
- except Exception as e:
33
- print(f"DHIB AAD U WEYN: Lama soo dejin karin moodeelka. Khaladku waa: {e}")
34
- # App-ku wuu shaqayn doonaa laakiin wuxuu soo celin doonaa khalad endpoint-ka transcribe
35
 
 
 
 
 
 
 
 
 
36
  # --- API Endpoints ---
37
 
38
  @app.get("/")
 
20
  # --- Soo Dejinta Moodeelka (Model Loading) ---
21
  # Kani hadda waxa uu isticmaali doonaa jidka keydka (cache path) ee lagu dejiyay Dockerfile-ka
22
  # oo ah /app/hf-cache, kaas oo leh ruqsadaha saxda ah.
23
+ # Faylkan waxa la isticmaalayaa oo kaliya inta lagu jiro dhismaha Docker
24
+ # si loo soo dejiyo moodeelka loogana fogaado khaladaadka ruqsadaha ee runtime-ka
25
+
26
+ from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
27
+ import os
28
+
29
  MODEL_ID = "Mustafaa4a/ASR-Somali"
 
 
 
 
 
 
 
 
 
 
 
30
 
31
+ print(f"Waxaa la bilaabayaa soo dejinta moodeelka: {MODEL_ID}")
32
+ print(f"Lagu keydin doonaa galka: {os.environ.get('HF_HOME')}")
33
+
34
+ # Labadan sadar ayaa kicin doona soo dejinta
35
+ processor = Wav2Vec2Processor.from_pretrained(MODEL_ID)
36
+ model = Wav2Vec2ForCTC.from_pretrained(MODEL_ID)
37
+
38
+ print("Soo dejinta moodeelka waa la dhammeystiray.")
39
  # --- API Endpoints ---
40
 
41
  @app.get("/")