Update app.py

app.py

@@ -639,12 +639,6 @@ def convert_ebook_to_audio(ebook_file, target_voice_file, language, use_custom_m
     # If the language argument is set use it instead
     language = args.language if args.language else language
 
-    if use_custom_model and custom_model_file and custom_config_file and custom_vocab_file:
-        custom_model = {
-            'model': custom_model_file.name,
-            'config': custom_config_file.name,
-            'vocab': custom_vocab_file.name
-        }
     # If headless is used with the custom model arguments
     if args.use_custom_model and args.custom_model and args.custom_config and args.custom_vocab:
         custom_model = {
@@ -652,6 +646,13 @@ def convert_ebook_to_audio(ebook_file, target_voice_file, language, use_custom_m
             'config': args.custom_config,
             'vocab': args.custom_vocab
         }
+
+    elif use_custom_model and custom_model_file and custom_config_file and custom_vocab_file:
+        custom_model = {
+            'model': custom_model_file.name,
+            'config': custom_config_file.name,
+            'vocab': custom_vocab_file.name
+        }
     if (use_custom_model and custom_model_url) or (args.use_custom_model and custom_model_url):
         print(f"Received custom model URL: {custom_model_url}")
         download_dir = os.path.join(".", "Working_files", "custom_model")