Florin Bobiș commited on
Commit
e5be17c
1 Parent(s): 075f690
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -30,7 +30,7 @@ def convert_to_mp3(video_file):
30
  def get_model():
31
  model_size = "large-v3"
32
  os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
33
- model = WhisperModel(model_size, device="cuda", compute_type="float16")
34
  return model
35
 
36
 
@@ -157,10 +157,10 @@ def initialize_app():
157
  if not "reel_id" in st.session_state:
158
  st.session_state.reel_id = None
159
 
160
- if not os.path.exists("temp"):
161
- os.makedirs("temp")
162
- if not os.path.exists("data"):
163
- os.makedirs("data")
164
 
165
  def init_ig() -> Client:
166
  cl = Client()
 
30
  def get_model():
31
  model_size = "large-v3"
32
  os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
33
+ model = WhisperModel(model_size, device="auto", compute_type="float16")
34
  return model
35
 
36
 
 
157
  if not "reel_id" in st.session_state:
158
  st.session_state.reel_id = None
159
 
160
+ if not os.path.exists("./temp"):
161
+ os.makedirs("./temp")
162
+ if not os.path.exists("./data"):
163
+ os.makedirs("./data")
164
 
165
  def init_ig() -> Client:
166
  cl = Client()