save

app.py

@@ -1,21 +1,22 @@
 import os
-os.system("pip install git+https://github.com/openai/whisper.git")
+os.system("pip install git+https://github.com/sanchit-gandhi/whisper-jax.git")
 
 import whisper
 from flask import Flask, jsonify, request
 import requests
 import time
 from transformers import pipeline
-from datasets import load_dataset
+from whisper_jax import FlaxWhisperPipline
 
 
 model = whisper.load_model("small")
-pipe = pipeline(
-    "automatic-speech-recognition",
-    model="openai/whisper-small.en",
-    chunk_length_s=15,
-    device=model.device,
-)
+# pipe = pipeline(
+#     "automatic-speech-recognition",
+#     model="openai/whisper-small.en",
+#     chunk_length_s=15,
+#     device=model.device,
+# )
+pipe = FlaxWhisperPipline("openai/whisper-large-v2")
 
 app = Flask(__name__)
 app.config['TIMEOUT'] = 60 * 10 # 10 mins