selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix imports

Browse Source
This commit is contained in:
Gokul Mohanarangan
2023-07-11 12:52:58 +05:30
parent 8deeff588f
commit 54de3683b3
8 changed files with 8 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
constants.py
View File

3
server_executor_cleaned.py
View File

@@ -1,8 +1,6 @@
import asyncio import asyncio
import datetime
import io import io
import json import json
import sys
import uuid import uuid
import wave import wave
from concurrent.futures import ThreadPoolExecutor from concurrent.futures import ThreadPoolExecutor
@@ -14,6 +12,7 @@ from aiortc.contrib.media import MediaRelay
from av import AudioFifo from av import AudioFifo
from loguru import logger from loguru import logger
from whisper_jax import FlaxWhisperPipline from whisper_jax import FlaxWhisperPipline
from utils.run_utils import run_in_executor from utils.run_utils import run_in_executor
pcs = set() pcs = set()

5
server_multithreaded.py
View File

@@ -1,9 +1,7 @@
import asyncio import asyncio
from utils.run_utils import config
import datetime import datetime
import io import io
import json import json
import os
import threading import threading
import uuid import uuid
import wave import wave
@@ -16,8 +14,9 @@ from aiortc.contrib.media import MediaRelay
from av import AudioFifo from av import AudioFifo
from sortedcontainers import SortedDict from sortedcontainers import SortedDict
from whisper_jax import FlaxWhisperPipline from whisper_jax import FlaxWhisperPipline
from utils.log_utils import logger from utils.log_utils import logger
from utils.run_utils import Mutex from utils.run_utils import config, Mutex
WHISPER_MODEL_SIZE = config['DEFAULT']["WHISPER_REAL_TIME_MODEL_SIZE"] WHISPER_MODEL_SIZE = config['DEFAULT']["WHISPER_REAL_TIME_MODEL_SIZE"]
pcs = set() pcs = set()

3
stream_client.py
View File

@@ -1,6 +1,5 @@
import ast import ast
import asyncio import asyncio
from utils.run_utils import config
import time import time
import uuid import uuid
@@ -12,7 +11,7 @@ from aiortc import (RTCPeerConnection, RTCSessionDescription)
from aiortc.contrib.media import (MediaPlayer, MediaRelay) from aiortc.contrib.media import (MediaPlayer, MediaRelay)
from utils.log_utils import logger from utils.log_utils import logger
from utils.run_utils import Mutex from utils.run_utils import config, Mutex
file_lock = Mutex(open("test_sm_6.txt", "a")) file_lock = Mutex(open("test_sm_6.txt", "a"))

4
utils/file_utils.py
View File

@@ -1,10 +1,10 @@
import configparser
import sys import sys
import boto3 import boto3
import botocore import botocore
from run_utils import config
from log_utils import logger from log_utils import logger
from run_utils import config
BUCKET_NAME = config["DEFAULT"]["BUCKET_NAME"] BUCKET_NAME = config["DEFAULT"]["BUCKET_NAME"]

6
utils/text_utilities.py
View File

@@ -1,5 +1,3 @@
import configparser
import nltk import nltk
import torch import torch
from nltk.corpus import stopwords from nltk.corpus import stopwords
@@ -7,13 +5,13 @@ from nltk.tokenize import word_tokenize
from sklearn.feature_extraction.text import TfidfVectorizer from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import cosine_similarity from sklearn.metrics.pairwise import cosine_similarity
from transformers import BartForConditionalGeneration, BartTokenizer from transformers import BartForConditionalGeneration, BartTokenizer
from run_utils import config
from log_utils import logger from log_utils import logger
from run_utils import config
nltk.download('punkt', quiet=True) nltk.download('punkt', quiet=True)
def preprocess_sentence(sentence): def preprocess_sentence(sentence):
stop_words = set(stopwords.words('english')) stop_words = set(stopwords.words('english'))
tokens = word_tokenize(sentence.lower()) tokens = word_tokenize(sentence.lower())

1
whisjax.py
View File

@@ -5,7 +5,6 @@
# summarize podcast.mp3 summary.txt # summarize podcast.mp3 summary.txt
import argparse import argparse
import configparser
import os import os
import re import re
import subprocess import subprocess

2
whisjax_realtime.py
View File

@@ -1,6 +1,5 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import configparser
import time import time
import wave import wave
from datetime import datetime from datetime import datetime
@@ -17,7 +16,6 @@ from utils.run_utils import config
from utils.text_utilities import post_process_transcription, summarize from utils.text_utilities import post_process_transcription, summarize
from utils.viz_utilities import create_talk_diff_scatter_viz, create_wordcloud from utils.viz_utilities import create_talk_diff_scatter_viz, create_wordcloud
WHISPER_MODEL_SIZE = config['DEFAULT']["WHISPER_MODEL_SIZE"] WHISPER_MODEL_SIZE = config['DEFAULT']["WHISPER_MODEL_SIZE"]
FRAMES_PER_BUFFER = 8000 FRAMES_PER_BUFFER = 8000