SarahMakk commited on
Commit
273b099
·
verified ·
1 Parent(s): 882d1dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import streamlit as st
2
  from transformers import pipeline
3
 
4
  # Load the Hugging Face pipelines
5
- classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
6
- sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
7
 
8
  # Define the categories for customer feedback
9
  CATEGORIES = ["Pricing", "Feature", "Customer Service", "Delivery", "Quality"]
 
2
  from transformers import pipeline
3
 
4
  # Load the Hugging Face pipelines
5
+ classifier = pipeline("zero-shot-classification", model="MoritzLaurer/DeBERTa-v3-base-mnli-fever-anli")
6
+ sentiment_analyzer = pipeline("sentiment-analysis", model="moshew/bert-tiny-sst2-distilled")
7
 
8
  # Define the categories for customer feedback
9
  CATEGORIES = ["Pricing", "Feature", "Customer Service", "Delivery", "Quality"]