t

app.py

@@ -15,7 +15,7 @@ from csm1b_dv import (
     split_sentences, extract_audio_output, generate_conversation,
     change_model_and_update_ui
 )
-from dia_1_6B_dv import get_dia_1_6B_tab
+#from dia_1_6B_dv import get_dia_1_6B_tab
 
 with gr.Blocks(
     title="Dhivehi (Thaana) Text-to-Speech",
@@ -36,7 +36,7 @@ with gr.Blocks(
     """
 ) as app:
     get_csm1b_tab()
-    get_dia_1_6B_tab()
+   # get_dia_1_6B_tab()
 
 if __name__ == "__main__":
     app.launch(share=False)

csm1b_dv.py

@@ -80,9 +80,13 @@ def load_model(model_name):
         return False
     model_id = MODELS[model_name]
     print(f"Model ID: {model_id}")
-    if current_model_id == model_id:
-        print(f"Model '{model_name}' already loaded")
+    if current_model_id == model_id and model is not None and processor is not None:
+        print(f"Model '{model_name}' already loaded and valid")
         return True
+    elif current_model_id == model_id:
+        print(f"Model '{model_name}' was previously loaded but model/processor is None, reloading...")
+        # Reset the current_model_id to force reload
+        current_model_id = None
     try:
         print(f"Loading model '{model_name}' on {device}...")
         if model is not None:
@@ -92,7 +96,7 @@ def load_model(model_name):
         
         print(f"Loading processor from {model_id}...")
         processor = AutoProcessor.from_pretrained(model_id)
-        print(f"Processor loaded successfully")
+        print(f"Processor loaded successfully: {processor is not None}")
         
         if hasattr(processor.tokenizer, "init_kwargs"):
             processor.tokenizer.init_kwargs.pop("pad_to_multiple_of", None)
@@ -103,7 +107,7 @@ def load_model(model_name):
             device_map=device,
             torch_dtype=torch.float32
         )
-        print(f"Model loaded successfully")
+        print(f"Model loaded successfully: {model is not None}")
         
         # Check if model has generate method
         if hasattr(model, 'generate'):