Spaces:
Running
Running
Muhammad Abdullah commited on
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)
|