Update app.py

app.py

@@ -4,11 +4,12 @@ import torch
 
 model_id = "moonshotai/Kimi-K2-Instruct"
 
-tokenizer = AutoTokenizer.from_pretrained(model_id)
+tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     torch_dtype=torch.float16,
-    device_map="auto"
+    device_map="auto",
+    trust_remote_code=True
 )
 
 def chat(prompt):
@@ -29,4 +30,4 @@ def chat(prompt):
     response = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
     return response.strip()
 
-gr.Interface(fn=chat, inputs=gr.Textbox(lines=5, label="Prompt"), outputs="text").launch()
+gr.Interface(fn=chat, inputs=gr.Textbox(lines=5, label="Prompt"), outputs="text").launch()