Update app.py

app.py

@@ -1,5 +1,25 @@
+from transformers import pipeline
 import gradio as gr
 
-demo = gr.load("MuntasirHossain/distilbert-finetuned-ag-news", src="models")
+model_checkpoint = "MuntasirHossain/distilbert-finetuned-ag-news"
 
-demo.launch()
+model = pipeline("text-classification", model=model_checkpoint)
+
+
+def predict(prompt):
+    completion = model(prompt)[0]["label"]
+    return completion
+
+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."], ["Business"]]
+
+gr.Interface(fn=predict,
+    inputs="textbox",
+    outputs="text",
+    title=title,
+    theme = theme,
+    description=description,
+    examples=examples,
+).launch()