Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
 #Defining the labels of the models
-labels = ["business", "science","health", "world", "sport", "politics", "entertainment", "tech"]
+labels = [ "business", "science", "health", "world", "sport", "politics", "entertainment", "tech", "education", "environment", "travel", "lifestyle", "crime", "opinion", "weather", "culture", "art", "food", "automotive", "finance", "international" ]
 
 #Defining the models and tokenuzer
 model_name = "valurank/finetuned-distilbert-news-article-categorization"
@@ -64,4 +64,4 @@ demo = gr.Interface(get_category, inputs=gr.Textbox(label="Drop your articles he
 
 #Launching the gradio app
 if __name__ == "__main__":
-  demo.launch(debug=True)
+  demo.launch(debug=True)