Update app.py
Browse files
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 |
|