Hugging Face's logo

Spaces:
ShubhamCoder01
/
stress-detector
Sleeping

App Files Community
ShubhamCoder01 commited on
Commit
ff7e370
·
verified ·
1 Parent(s): 198a8cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ physical_model = AutoModelForSequenceClassification.from_pretrained(physical_mod
19
  physical_id2label = {0: 'Joy', 1: 'Sadness', 2: 'Anger', 3: 'Love', 4: 'Surprise', 5: 'Fear'}
20
 
21
  # Polarity Model
22
- polarity_model_name = "ShubhamCoder01/polarity-model"
23
  polarity_tokenizer = AutoTokenizer.from_pretrained(polarity_model_name)
24
  polarity_model = AutoModelForSequenceClassification.from_pretrained(polarity_model_name)
25
  polarity_id2label = {0: 'NEGATIVE', 1: 'POSITIVE', 2:"NORMAL"}
 
19
  physical_id2label = {0: 'Joy', 1: 'Sadness', 2: 'Anger', 3: 'Love', 4: 'Surprise', 5: 'Fear'}
20
 
21
  # Polarity Model
22
+ polarity_model_name = "ShubhamCoder01/college-context-polarity-detect"
23
  polarity_tokenizer = AutoTokenizer.from_pretrained(polarity_model_name)
24
  polarity_model = AutoModelForSequenceClassification.from_pretrained(polarity_model_name)
25
  polarity_id2label = {0: 'NEGATIVE', 1: 'POSITIVE', 2:"NORMAL"}