SarahMakk commited on
Commit
80d4bd4
·
verified ·
1 Parent(s): 273b099

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ 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"]
 
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="SarahMakk/CustomModel_amazon_sentiment_moshew_128_10k")
7
 
8
  # Define the categories for customer feedback
9
  CATEGORIES = ["Pricing", "Feature", "Customer Service", "Delivery", "Quality"]