Update app.py
Browse files
app.py
CHANGED
@@ -107,11 +107,11 @@ def infer(user_audio) -> tuple[int, np.ndarray]:
|
|
107 |
generator = load_csm_1b(model_path, device)
|
108 |
logging.info("Sesame CSM 1B loaded successfully.")
|
109 |
|
110 |
-
whisper_model = whisper.load_model("small.en", device=device
|
111 |
logging.info("Whisper model loaded successfully.")
|
112 |
|
113 |
tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-3-1b-it")
|
114 |
-
model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it").to(device
|
115 |
logging.info("Gemma 3 1B pt model loaded successfully.")
|
116 |
|
117 |
if not user_audio:
|
|
|
107 |
generator = load_csm_1b(model_path, device)
|
108 |
logging.info("Sesame CSM 1B loaded successfully.")
|
109 |
|
110 |
+
whisper_model = whisper.load_model("small.en", device=device)
|
111 |
logging.info("Whisper model loaded successfully.")
|
112 |
|
113 |
tokenizer_gemma = AutoTokenizer.from_pretrained("google/gemma-3-1b-it")
|
114 |
+
model_gemma = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it").to(device)
|
115 |
logging.info("Gemma 3 1B pt model loaded successfully.")
|
116 |
|
117 |
if not user_audio:
|