Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
-from transformers import AutoTokenizer, PreTrainedTokenizer
+from transformers import AutoTokenizer, PreTrainedTokenizer, AutoConfig
 import torch
 from custom_model import CustomModel
 
-# Load the tokenizer and custom model
+# Load the model and tokenizer
 model_name = "deepseek-ai/DeepSeek-V3"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = CustomModel.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)
+model = CustomModel.from_pretrained(model_name, config=config, trust_remote_code=True)
 
 def classify_text(text):
     inputs = tokenizer(text, return_tensors="pt")