Spaces:
Runtime error
Runtime error
update app.py
Browse files
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 |
-
|
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):
|