ahm14 commited on
Commit
f83098d
·
verified ·
1 Parent(s): 76c1dd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from io import BytesIO
13
  # pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
14
 
15
  # Set up AI models
16
- emotion_classifier = pipeline("text-classification", model="joeddav/distilbert-base-uncased-go-emotions")
17
  keyword_extractor = KeyBERT()
18
  zero_shot_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
19
 
 
13
  # pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
14
 
15
  # Set up AI models
16
+ emotion_classifier = pipeline("text-classification", model="joeddav/distilbert-base-uncased-go-emotions-student")
17
  keyword_extractor = KeyBERT()
18
  zero_shot_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
19