Spaces:
Sleeping
Sleeping
Commit
·
89c88e1
1
Parent(s):
0e3b935
inference API
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ LOCAL_MODEL = Path(__file__).resolve().parents[1] / "models" / "incident_classif
|
|
10 |
|
11 |
# Category classifier (your fine-tuned model if available)
|
12 |
if LOCAL_MODEL.exists():
|
13 |
-
incident_classifier = pipeline("text-classification", model=
|
14 |
else:
|
15 |
incident_classifier = pipeline("text-classification", model="cardiffnlp/twitter-xlm-roberta-base")
|
16 |
|
|
|
10 |
|
11 |
# Category classifier (your fine-tuned model if available)
|
12 |
if LOCAL_MODEL.exists():
|
13 |
+
incident_classifier = pipeline("text-classification", model="brijeshpandya/incident-classifier")
|
14 |
else:
|
15 |
incident_classifier = pipeline("text-classification", model="cardiffnlp/twitter-xlm-roberta-base")
|
16 |
|