Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -17,9 +17,9 @@ if 'srt_content' not in st.session_state:
|
|
17 |
|
18 |
@st.cache_resource
|
19 |
def load_model():
|
20 |
-
model_id = "openai/whisper-
|
21 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
22 |
-
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=
|
23 |
).to(device)
|
24 |
processor = AutoProcessor.from_pretrained(model_id)
|
25 |
pipe = pipeline(
|
|
|
17 |
|
18 |
@st.cache_resource
|
19 |
def load_model():
|
20 |
+
model_id = "openai/whisper-tiny"
|
21 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
22 |
+
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=False, use_safetensors=True
|
23 |
).to(device)
|
24 |
processor = AutoProcessor.from_pretrained(model_id)
|
25 |
pipe = pipeline(
|