update app.py

app.py

@@ -1,10 +1,10 @@
 import torch
 import gradio as gr
 from transformers import AutoConfig
-from models import CustomClassifier
+from models import CustomClassifier, CustomClassificationConfig
 
 MODEL_ID = "yhamidullah/custom-classifier-demo"
-#config = AutoConfig.from_pretrained(MODEL_ID)
+config = CustomClassificationConfig.from_pretrained(MODEL_ID)
 model = CustomClassifier.from_pretrained(MODEL_ID)
 model.eval()