Update app.py
Browse files
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("/")
|