Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ YT_LENGTH_LIMIT_S = 3600 # limit to 1 hour YouTube files
|
|
17 |
|
18 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
19 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
20 |
-
model_id = "Kushtrim/whisper-large-v3-turbo-shqip"
|
21 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
22 |
model_id, torch_dtype=torch_dtype, use_safetensors=True, token=True).to(device)
|
23 |
processor = AutoProcessor.from_pretrained(model_id, token=True)
|
|
|
17 |
|
18 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
19 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
20 |
+
model_id = "Kushtrim/whisper-large-v3-turbo-shqip-50h"
|
21 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(
|
22 |
model_id, torch_dtype=torch_dtype, use_safetensors=True, token=True).to(device)
|
23 |
processor = AutoProcessor.from_pretrained(model_id, token=True)
|