Spaces:
Runtime error
Runtime error
Commit
·
43b366f
1
Parent(s):
277dc51
added Whisper model api
Browse files- app.py +2 -2
- whisper_app.py +0 -1
app.py
CHANGED
@@ -7,7 +7,7 @@ from langchain.vectorstores import FAISS
|
|
7 |
from langchain.chains import RetrievalQA
|
8 |
from langchain.prompts import PromptTemplate
|
9 |
from langchain.docstore.document import Document
|
10 |
-
|
11 |
import llm_ops
|
12 |
|
13 |
FILE_EXT = ['wav','mp3']
|
@@ -55,7 +55,7 @@ def process_documents(documents,data_chunk=1500,chunk_overlap=100):
|
|
55 |
def audio_processor(wav_file,API_key,wav_model='small',llm='HuggingFace',temperature=0.1,max_tokens=4096):
|
56 |
device='cpu'
|
57 |
logger.info("Loading Whsiper Model || Model size:{}".format(wav_model))
|
58 |
-
whisper = WHISPERModel(model_name=wav_model,device=device)
|
59 |
text_info = whisper.speech_to_text(audio_path=wav_file)
|
60 |
|
61 |
metadata = {"source": f"{wav_file}","duration":text_info['duration'],"language":text_info['language']}
|
|
|
7 |
from langchain.chains import RetrievalQA
|
8 |
from langchain.prompts import PromptTemplate
|
9 |
from langchain.docstore.document import Document
|
10 |
+
import whisper_app
|
11 |
import llm_ops
|
12 |
|
13 |
FILE_EXT = ['wav','mp3']
|
|
|
55 |
def audio_processor(wav_file,API_key,wav_model='small',llm='HuggingFace',temperature=0.1,max_tokens=4096):
|
56 |
device='cpu'
|
57 |
logger.info("Loading Whsiper Model || Model size:{}".format(wav_model))
|
58 |
+
whisper = whisper_app.WHISPERModel(model_name=wav_model,device=device)
|
59 |
text_info = whisper.speech_to_text(audio_path=wav_file)
|
60 |
|
61 |
metadata = {"source": f"{wav_file}","duration":text_info['duration'],"language":text_info['language']}
|
whisper_app.py
CHANGED
@@ -5,7 +5,6 @@ from whisper.audio import SAMPLE_RATE
|
|
5 |
from tenacity import retry, wait_random
|
6 |
import openai
|
7 |
import requests
|
8 |
-
import time
|
9 |
# os.environ['OPENAI_API_KEY'] = "sk-<API KEY>"
|
10 |
|
11 |
class WHISPERModel:
|
|
|
5 |
from tenacity import retry, wait_random
|
6 |
import openai
|
7 |
import requests
|
|
|
8 |
# os.environ['OPENAI_API_KEY'] = "sk-<API KEY>"
|
9 |
|
10 |
class WHISPERModel:
|