Samarth991 commited on
Commit
43b366f
1 Parent(s): 277dc51

added Whisper model api

Browse files
Files changed (2) hide show
  1. app.py +2 -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
- from whisper_app import WHISPERModel
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: