Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ label_map={
|
|
12 |
'LABEL_4':'fear',
|
13 |
'LABEL_5':'surprise'
|
14 |
}
|
15 |
-
pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer
|
16 |
|
17 |
def run_inference(text):
|
18 |
emotion = label_map[pipe(text)[0]['label']]
|
|
|
12 |
'LABEL_4':'fear',
|
13 |
'LABEL_5':'surprise'
|
14 |
}
|
15 |
+
pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)
|
16 |
|
17 |
def run_inference(text):
|
18 |
emotion = label_map[pipe(text)[0]['label']]
|