Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -15,10 +15,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
15
 
16
  load_dotenv()
17
 
18
- MODEL_URL = "TostAI/nsfw-text-detection-large"
19
- CLASS_NAMES = {0: "βœ… SAFE", 1: "⚠️ QUESTIONABLE", 2: "🚫 UNSAFE"}
20
- tokenizer = AutoTokenizer.from_pretrained(MODEL_URL)
21
- model = AutoModelForSequenceClassification.from_pretrained(MODEL_URL)
22
 
23
  class SessionManager:
24
  _instances = {}
 
15
 
16
  load_dotenv()
17
 
18
+
 
 
 
19
 
20
  class SessionManager:
21
  _instances = {}