everydayumar commited on
Commit
0cb4d50
·
verified ·
1 Parent(s): c82eeaa

Update app.py

Browse files
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 = {}