makanaan commited on
Commit
20f27a0
·
1 Parent(s): 31032b6

update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -2,15 +2,15 @@ from transformers import pipeline
2
  import gradio as gr
3
 
4
  from transformers import AutoTokenizer, AutoModelWithLMHead
5
- from os import path
6
 
7
- MODEL_DIRECTORY = "/models/mrm8488-t5-base-finetuned-emotion"
8
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion", use_fast=False)
9
- if not path.exists(MODEL_DIRECTORY):
10
- model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-emotion")
11
- model.save_pretrained(MODEL_DIRECTORY)
12
- else:
13
- model = AutoModelWithLMHead.from_pretrained(MODEL_DIRECTORY)
14
 
15
 
16
  def get_emotion(text):
 
2
  import gradio as gr
3
 
4
  from transformers import AutoTokenizer, AutoModelWithLMHead
5
+ #from os import path
6
 
7
+ #MODEL_DIRECTORY = "/models/mrm8488-t5-base-finetuned-emotion"
8
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-emotion", use_fast=False)
9
+ #if not path.exists(MODEL_DIRECTORY):
10
+ model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-emotion")
11
+ # model.save_pretrained(MODEL_DIRECTORY)
12
+ #else:
13
+ # model = AutoModelWithLMHead.from_pretrained(MODEL_DIRECTORY)
14
 
15
 
16
  def get_emotion(text):