GogetaBlueMUI commited on
Commit
84e6fa4
·
verified ·
1 Parent(s): 286f91d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -293,8 +293,8 @@ class TranscriptionProgress:
293
  def load_model(language='en', summarizer_type='bart'):
294
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
295
  if language == 'ur':
296
- processor = AutoProcessor.from_pretrained("GogetaBlueMUI/whisper-medium-ur-fleurs")
297
- model = AutoModelForSpeechSeq2Seq.from_pretrained("GogetaBlueMUI/whisper-medium-ur-fleurs").to(device)
298
  else:
299
  processor = AutoProcessor.from_pretrained("openai/whisper-small")
300
  model = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-small").to(device)
 
293
  def load_model(language='en', summarizer_type='bart'):
294
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
295
  if language == 'ur':
296
+ processor = AutoProcessor.from_pretrained("GogetaBlueMUI/whisper-medium-ur-fleurs-v2")
297
+ model = AutoModelForSpeechSeq2Seq.from_pretrained("GogetaBlueMUI/whisper-medium-ur-fleurs-v2").to(device)
298
  else:
299
  processor = AutoProcessor.from_pretrained("openai/whisper-small")
300
  model = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-small").to(device)