Update app.py

app.py

@@ -164,29 +164,9 @@ Have a conversation with an AI using your reference voice!
 """)
 
     if not USING_SPACES:
-        load_chat_model_btn = gr.Button("Load Chat Model", variant="primary")
         chat_interface_container = gr.Column(visible=False)
 
-        @gpu_decorator
-        def load_chat_model():
-            global chat_model_state, chat_tokenizer_state
-            if chat_model_state is None:
-                gr.Info("Loading chat model...")
-                model_name = "deepseek-ai/Janus-Pro-7B"
-                chat_model_state = AutoModelForCausalLM.from_pretrained(
-                    model_name, device_map="auto"
-                )
-                chat_tokenizer_state = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-                gr.Info("Chat model loaded.")
-            return gr.update(visible=False), gr.update(visible=True)
-
-        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/Janus-Pro-7B"
-        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)
-
+        
     with chat_interface_container:
         with gr.Row():
             with gr.Column():