Spaces:

rbcurzon
/

speech-to-text

Running

rbcurzon commited on Mar 13

Commit

83733e7

verified ·

1 Parent(s): f79d643

Update app.py

Change model that can understand transcribe Filipino and Cebuano

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from transformers import WhisperProcessor, WhisperForConditionalGeneration, pipe
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 # load model and processor
-model_id = "rbcurzon/whisper-small-ceb"
 pipe = pipeline("automatic-speech-recognition", model=model_id, device=device)
 """**FastAPI**"""
@@ -29,7 +29,7 @@ from google.genai import types
 client = genai.Client(api_key=os.environ.get("GENAI_API_KEY")) # Do not share api key
 def translate(text, srcLang, tgtLang):
-    sys_instruct = "You are a professional translator. Return nothing but the translation only."
     response = client.models.generate_content(
         model="gemini-2.0-flash",
         config=types.GenerateContentConfig(

 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 # load model and processor
+model_id = "rbcurzon/whisper-small-fil"
 pipe = pipeline("automatic-speech-recognition", model=model_id, device=device)
 """**FastAPI**"""
 client = genai.Client(api_key=os.environ.get("GENAI_API_KEY")) # Do not share api key
 def translate(text, srcLang, tgtLang):
+    sys_instruct = "You are a professional translator. Do not give explanation."
     response = client.models.generate_content(
         model="gemini-2.0-flash",
         config=types.GenerateContentConfig(