Update app.py

app.py

@@ -6,16 +6,16 @@ model_checkpoint = "MuntasirHossain/distilbert-finetuned-ag-news"
 model = pipeline("text-classification", model=model_checkpoint)
 
 
-def predict(prompt):
-    completion = model(prompt)[0]["label"]
-    return completion
+def classify(text):
+    label = model(text)[0]["label"]
+    return label
 
 description = "This AI model is trained to classify news articles into four categories: World, Sports, Business and Science/Tech." 
 title = "Classify Your Articles"
 theme = "peach"
 examples=[["Global Retail Giants Gear Up for Record-Breaking Holiday Sales Season Amidst Supply Chain Challenges and Rising Consumer Demand."]]
 
-gr.Interface(fn=predict,
+gr.Interface(fn=classify,
     inputs="textbox",
     outputs="text",
     title=title,