Hugging Face's logo

Spaces:
neurotech
/
Swahili-sentiment-analysis
Runtime error

App Files Community
1
kalebu commited on
Commit
32fafb4
·
1 Parent(s): e3128c9

added precise index on the prediction

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ model = joblib.load("model.pkl")
6
 
7
 
8
  def get_sentiment(input_text):
9
- return model.predict([input_text])
10
 
11
 
12
  iface = gr.Interface(
 
6
 
7
 
8
  def get_sentiment(input_text):
9
+ return model.predict([input_text][0].capitalize())
10
 
11
 
12
  iface = gr.Interface(