Himanshusingh
commited on
Commit
•
47f38d5
1
Parent(s):
03e8974
Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,17 @@
|
|
1 |
import gradio
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
|
3 |
def my_inference_function(name):
|
4 |
return "Hello " + name + "!"
|
|
|
1 |
import gradio
|
2 |
+
import nltk
|
3 |
+
import pandas as pd
|
4 |
+
from transformers import pipeline
|
5 |
+
|
6 |
+
summarizer = pipeline('summarization', model='t5-base')
|
7 |
+
|
8 |
+
# classifier_model_name = 'bhadresh-savani/distilbert-base-uncased-emotion'
|
9 |
+
# classifier_emotions = ['anger', 'disgust', 'fear', 'joy', 'sadness', 'surprise']
|
10 |
+
|
11 |
+
classifier_model_name = 'ProsusAI/finbert'
|
12 |
+
classifier_emotions = ['positive', 'neutral', 'negative']
|
13 |
+
|
14 |
+
classifier = pipeline('text-classification', model=classifier_model_name)
|
15 |
|
16 |
def my_inference_function(name):
|
17 |
return "Hello " + name + "!"
|