Spaces:
Runtime error
Runtime error
Florin Bobiș
commited on
Commit
•
e5be17c
1
Parent(s):
075f690
test
Browse files
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="
|
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()
|