mike23415 commited on
Commit
fb2a718
·
verified ·
1 Parent(s): e3da27a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,8 +15,8 @@ CORS(app)
15
  sock = Sock(app)
16
 
17
  # AI classification pipeline using Facebook's zero-shot model (lightweight for free-tier)
18
- classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
19
 
 
20
  # Active chat sessions: {session_id: [ {role, msg, ip, timestamp} ]}
21
  SESSIONS = {}
22
 
 
15
  sock = Sock(app)
16
 
17
  # AI classification pipeline using Facebook's zero-shot model (lightweight for free-tier)
 
18
 
19
+ classifier = pipeline("zero-shot-classification", model="valhalla/distilbart-mnli-12-1")
20
  # Active chat sessions: {session_id: [ {role, msg, ip, timestamp} ]}
21
  SESSIONS = {}
22