Update app.py

app.py

@@ -150,7 +150,7 @@ Have a conversation with an AI using your reference voice!
             global chat_model_state, chat_tokenizer_state
             if chat_model_state is None:
                 gr.Info("Loading chat model...")
-                model_name = "deepseek-ai/DeepSeek-V3"
+                model_name = "google/gemma-2-2b-it"
                 chat_model_state = AutoModelForCausalLM.from_pretrained(
                     model_name, device_map="auto"
                 )
@@ -161,7 +161,7 @@ Have a conversation with an AI using your reference voice!
         load_chat_model_btn.click(load_chat_model, outputs=[load_chat_model_btn, chat_interface_container])
     else:
         chat_interface_container = gr.Column()
-        model_name = "deepseek-ai/DeepSeek-V3"
+        model_name = "google/gemma-2-2b-it"
         chat_model_state = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto", trust_remote_code=True)
         chat_tokenizer_state = AutoTokenizer.from_pretrained(model_name)