f

app.py

@@ -1,19 +1,10 @@
 import gradio as gr
-from transformers import AutoTokenizer
-from transformers.utils import logging
+from transformers import AutoTokenizer, AutoModel
 
-# Enable logging to see debug messages
-logging.set_verbosity_info()
-
-# Import custom configuration and model classes
-from transformers_modules.deepseek_ai.DeepSeek_R1.configuration_deepseek import DeepseekV3Config
-from transformers_modules.deepseek_ai.DeepSeek_R1.modeling_deepseek import DeepseekV3Model
-
-# Load model and tokenizer
+# Load the model and tokenizer with `trust_remote_code=True`
 model_name = "deepseek-ai/DeepSeek-R1"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-config = DeepseekV3Config.from_pretrained(model_name, trust_remote_code=True)
-model = DeepseekV3Model.from_pretrained(model_name, config=config, trust_remote_code=True)
+model = AutoModel.from_pretrained(model_name, trust_remote_code=True)
 
 def classify_text(input_text):
     # Tokenize input