Update app.py

app.py

@@ -36,19 +36,19 @@ def classify_compliant(text):
   ranking = ranking[::-1]
 
   for i in range(len(scores)):
-    l = config.id2label[ranking[i]]
-    s = scores[ranking[i]]
-    probs[l] = np.round(float(s), 4)
-  return probs
+    l = config.id2label[ranking[0]]
+    #s = scores[ranking[i]]
+    #probs[l] = np.round(float(s), 4)
+  return l
 
 
 #build the Gradio app
 #Instructuction = "Write an imaginary review about a product or service you might be interested in."
-title="Text Sentiment Analysis"
-description = """Write a Good or Bad review about an imaginary product or service,\
-   see how the machine learning model is able to predict your sentiments"""
+title="Consumer Complaint Segmentation"
+description = """Write a complaint insurance product or service,\
+   see how the machine learning model is able to predict your Complaint type"""
 article = """
-            - Click submit button to test sentiment analysis prediction
+            - Click submit button to test Consumer Complaint Segmentation
             - Click clear button to refresh text
            """