cptsubtext commited on
Commit
eecdc1c
·
1 Parent(s): cb5bbce

calculation for free tier size adjusted

Browse files
Files changed (2) hide show
  1. app.py +2 -3
  2. 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