switch automodel for automodelforseqclass

main.py

@@ -3,7 +3,7 @@ import gradio as gr
 from fastapi import FastAPI
 import uvicorn
 from pydantic import BaseModel
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 app = FastAPI(title="EmCoder API & UI")
 
@@ -11,7 +11,9 @@ app = FastAPI(title="EmCoder API & UI")
 repo_id = "yezdata/EmCoder"
 
 tokenizer = AutoTokenizer.from_pretrained("answerdotai/ModernBERT-base")
-model = AutoModel.from_pretrained(repo_id, trust_remote_code=True)
+model = AutoModelForSequenceClassification.from_pretrained(
+    repo_id, trust_remote_code=True
+)
 
 model.eval()