Kushtrim commited on
Commit
84eb3d0
1 Parent(s): 560d03e

Update app.py

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