From 19dfb1d0272af4c89eb4a3dc1629c4aa85266078 Mon Sep 17 00:00:00 2001
From: Gokul Mohanarangan <gokul@monadical.com>
Date: Wed, 20 Sep 2023 20:02:52 +0530
Subject: [PATCH] Upgrade to a bigger translation model

---
 server/gpu/modal/reflector_transcriber.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/server/gpu/modal/reflector_transcriber.py b/server/gpu/modal/reflector_transcriber.py
index ff4caff7..df24e58f 100644
--- a/server/gpu/modal/reflector_transcriber.py
+++ b/server/gpu/modal/reflector_transcriber.py
@@ -6,7 +6,6 @@ Reflector GPU backend - transcriber
 import os
 import tempfile
 
-import modal
 from modal import Image, Secret, Stub, asgi_app, method
 from pydantic import BaseModel
 
@@ -16,7 +15,7 @@ WHISPER_COMPUTE_TYPE: str = "float16"
 WHISPER_NUM_WORKERS: int = 1
 
 # Translation Model
-TRANSLATION_MODEL = "facebook/m2m100_418M"
+TRANSLATION_MODEL = "facebook/m2m100_1.2B"
 
 IMAGE_MODEL_DIR = "/root/transcription_models"