Update app.py

app.py

@@ -95,8 +95,8 @@ def load_model(model_path, progress=gr.Progress()):
         config = AutoConfig.from_pretrained(model_path, trust_remote_code=True)
         if 'MoE' in model_path:
             config_moe = config
-            config_moe["auto_map"]["AutoConfig"] = "./configuration_upcycling_qwen2_moe.UpcyclingQwen2MoeConfig"
-            config_moe["auto_map"]["AutoModelForCausalLM"] = "./modeling_upcycling_qwen2_moe.UpcyclingQwen2MoeForCausalLM"
+            config_moe.auto_map["AutoConfig"] = "./configuration_upcycling_qwen2_moe.UpcyclingQwen2MoeConfig"
+            config_moe.auto_map["AutoModelForCausalLM"] = "./modeling_upcycling_qwen2_moe.UpcyclingQwen2MoeForCausalLM"
         current_tokenizer = AutoTokenizer.from_pretrained(model_path, use_fast=False,trust_remote_code=True)
         
         progress(0.5, desc="Loading model...")