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 + "!"
|