Spaces:
Sleeping
Sleeping
cptsubtext
commited on
Commit
·
eecdc1c
1
Parent(s):
cb5bbce
calculation for free tier size adjusted
Browse files- app.py +2 -3
- requirements.txt +1 -2
app.py
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
import streamlit as st
|
2 |
from stable_whisper import load_model
|
3 |
from stable_whisper import load_hf_whisper
|
4 |
-
import ffmpeg
|
5 |
import requests
|
6 |
import os
|
7 |
|
@@ -19,14 +18,14 @@ api_token = st.text_input("API Token (Unlimited)")
|
|
19 |
model_size = st.selectbox("Model Size", ("tiny", "base", "small", "medium"))
|
20 |
|
21 |
def transcribe_to_subtitle(audio_bytes, model_name):
|
22 |
-
st.error(ffmpeg.probe('audio_bytes')['format']['duration'])
|
23 |
"""Transcribe audio to subtitle using OpenAI Whisper"""
|
24 |
# Load model based on selection
|
25 |
model = load_model(model_name)
|
26 |
#speedmodel = load_hf_whisper(model_name)
|
27 |
|
28 |
# Check file size for free tier
|
29 |
-
if use_free_tier and len(audio_bytes) > 2 * 60 * 1024:
|
|
|
30 |
st.error("Free tier only supports audio files under 2 minutes")
|
31 |
return
|
32 |
|
|
|
1 |
import streamlit as st
|
2 |
from stable_whisper import load_model
|
3 |
from stable_whisper import load_hf_whisper
|
|
|
4 |
import requests
|
5 |
import os
|
6 |
|
|
|
18 |
model_size = st.selectbox("Model Size", ("tiny", "base", "small", "medium"))
|
19 |
|
20 |
def transcribe_to_subtitle(audio_bytes, model_name):
|
|
|
21 |
"""Transcribe audio to subtitle using OpenAI Whisper"""
|
22 |
# Load model based on selection
|
23 |
model = load_model(model_name)
|
24 |
#speedmodel = load_hf_whisper(model_name)
|
25 |
|
26 |
# Check file size for free tier
|
27 |
+
if use_free_tier and len(audio_bytes) > 0.048 * 2 * 60 * 1024:
|
28 |
+
st.error(audio_bytes)
|
29 |
st.error("Free tier only supports audio files under 2 minutes")
|
30 |
return
|
31 |
|
requirements.txt
CHANGED
@@ -1,3 +1,2 @@
|
|
1 |
openai-whisper
|
2 |
-
stable-ts
|
3 |
-
ffmpeg-python
|
|
|
1 |
openai-whisper
|
2 |
+
stable-ts
|
|