Update app.py

app.py

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