save

app.py

@@ -6,11 +6,13 @@ from flask import Flask, jsonify, request
 import requests
 import time
 from transformers import pipeline
+from datasets import load_dataset
 
-model = whisper.load_model("small.en")
+
+model = whisper.load_model("small")
 pipe = pipeline(
     "automatic-speech-recognition",
-    model="openai/whisper-small",
+    model=model,
     chunk_length_s=15,
     device=model.device,
 )
@@ -51,6 +53,11 @@ def runApi():
     options = whisper.DecodingOptions(fp16 = False)
     result = whisper.decode(model, mel, options)
 
+    ds = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
+    sample = ds[0]["audio"]
+    prediction = pipe(sample.copy())["text"]
+    print(prediction)
+
     end_time = time.time()
     total_time = end_time - start_time