f

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
-# Load the 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)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+model = AutoModelForSequenceClassification.from_pretrained(model_name, trust_remote_code=True)
 
 def classify_text(input_text):
     # Tokenize the input