Himanshusingh commited on
Commit
47f38d5
1 Parent(s): 03e8974

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
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 + "!"