Spaces:
Running
Running
Update app.py
Browse files
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)
|