Spaces:
Sleeping
Sleeping
| # services/zero_shot.py | |
| from transformers import pipeline | |
| import streamlit as st | |
| def load_zero_shot_model(): | |
| return pipeline( | |
| "zero-shot-classification", | |
| model="facebook/bart-large-mnli" | |
| ) | |
| def classify_intent(text: str): | |
| classifier = load_zero_shot_model() | |
| labels = [ | |
| "sales", | |
| "lead generation", | |
| "brand awareness", | |
| "engagement", | |
| "traffic" | |
| ] | |
| result = classifier(text, labels) | |
| return { | |
| "label": result["labels"][0], | |
| "score": float(result["scores"][0]) | |
| } | |