pragnakalp commited on
Commit
0cc82f8
1 Parent(s): c6a2c72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ DATASET_REPO_ID = "pragnakalp/emotion_detection"
13
  print("is none?", HF_TOKEN is None)
14
 
15
  sentences_value = """Raj loves Simran.\nLast year I lost my Dog.\nI bought a new phone!\nShe is scared of cockroaches.\nWow! I was not expecting that.\nShe got mad at him."""
16
-
 
17
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion")
18
  model_base = AutoModelWithLMHead.from_pretrained(model_path)
19
 
 
13
  print("is none?", HF_TOKEN is None)
14
 
15
  sentences_value = """Raj loves Simran.\nLast year I lost my Dog.\nI bought a new phone!\nShe is scared of cockroaches.\nWow! I was not expecting that.\nShe got mad at him."""
16
+ cwd = os.getcwd()
17
+ model_path = os.path.join(cwd)
18
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion")
19
  model_base = AutoModelWithLMHead.from_pretrained(model_path)
20