umsee commited on
Commit
6595106
1 Parent(s): ea56466

text preprocessing added

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  from fastai.text.all import *
3
 
4
  #loading the model from pretrained weights
@@ -7,6 +8,11 @@ model = load_learner('emot.pkl')
7
  labels = ['sadnesss','joy','love','anger','fear','surprise']
8
 
9
  def make_prediction(text):
 
 
 
 
 
10
  choice,pos,preds= model.predict(text)
11
  return {labels[i]:float(preds[i]) for i in range(len(labels))}
12
 
 
1
  import gradio as gr
2
+ import string
3
  from fastai.text.all import *
4
 
5
  #loading the model from pretrained weights
 
8
  labels = ['sadnesss','joy','love','anger','fear','surprise']
9
 
10
  def make_prediction(text):
11
+ #text preprocessing
12
+ tr = str.maketrans('','',string.punctuation)
13
+ text=text.translate(tr).lower()
14
+
15
+ #making prediction by passing to model
16
  choice,pos,preds= model.predict(text)
17
  return {labels[i]:float(preds[i]) for i in range(len(labels))}
18