Update app.py

app.py

@@ -46,7 +46,7 @@ def predict_text(params):
         logits = pred.logits
         pred_label = tf.argmax(logits, axis=1).numpy()[0]
         
-        label = {1:'positive', 0:'negative'}
+        label = {0: 'BUSINESS', 1: 'COMEDY', 2: 'CRIME', 3: 'FOOD & DRINK', 4: 'POLITICS', 5: 'SPORTS', 6: 'TRAVEL'}
         result = {'text':text, 'label':label[pred_label]}
         solutions.append(result)