brijeshpandya commited on
Commit
89c88e1
·
1 Parent(s): 0e3b935

inference API

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=str(LOCAL_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