Update app.py

app.py

@@ -1,6 +1,9 @@
 import warnings
 import gradio as gr
-from transformers import AutoTokenizer, PreTrainedTokenizer, AutoConfig
+from transformers ```python
+import warnings
+import gradio as gr
+from transformers import AutoTokenizer, AutoConfig
 import torch
 from custom_model import CustomModel
 
@@ -21,7 +24,8 @@ print(f"Loading model from {model_name}...")
 model = CustomModel.from_pretrained(model_name, config=config, revision=revision, trust_remote_code=True)
 
 if model is None:
-    print("Failed to load model.")
+    print("Failed to load model. Exiting...")
+    exit(1)
 else:
     print("Model loaded successfully.")
 
@@ -40,10 +44,10 @@ def classify_text(text):
 try:
     iface = gr.Interface(
         fn=classify_text,
-        inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),  # Updated here
+        inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
         outputs=[
-            gr.Label(label="Predicted Class"),  # Updated here
-            gr.Label(label="Probabilities")  # Updated here
+            gr.Label(label="Predicted Class"),
+            gr.Label(label="Probabilities")
         ],
         title="DeepSeek-V3 Text Classification",
         description="Classify text using the DeepSeek-V3 model."