Update interfaces/classification.py

interfaces/classification.py

@@ -8,13 +8,9 @@ classification_examples = [
     [
         """
 "I recently purchased the Sony WH-1000XM4 Wireless Noise-Canceling Headphones from Amazon and I must say, I'm thoroughly impressed. The package arrived in New York within 2 days, thanks to Amazon Prime's expedited shipping.
-
 The headphones themselves are remarkable. The noise-canceling feature works like a charm in the bustling city environment, and the 30-hour battery life means I don't have to charge them every day. Connecting them to my Samsung Galaxy S21 was a breeze, and the sound quality is second to none.
-
 I also appreciated the customer service from Amazon when I had a question about the warranty. They responded within an hour and provided all the information I needed.
-
 However, the headphones did not come with a hard case, which was listed in the product description. I contacted Amazon, and they offered a 10% discount on my next purchase as an apology.
-
 Overall, I'd give these headphones a 4.5/5 rating and highly recommend them to anyone looking for top-notch quality in both product and service.""",
         "positive review, negative review, neutral review",
         0.5
@@ -32,9 +28,7 @@ Overall, I'd give these headphones a 4.5/5 rating and highly recommend them to a
         [
         """
 "Last night, we visited the new Italian restaurant downtown. The Margherita pizza was absolutely delightful, with a perfectly crisp crust and fresh basil. 
-
 However, the service was slow; it took over 20 minutes to take our order. The pasta arrived lukewarm, which was disappointing given the hype around this place. 
-
 On the bright side, the ambiance was cozy, and the wine selection was impressive. Overall, it was a mixed experience, but I might give it another try on a quieter evening."
         """,
         "Food Quality, Technology, Politics",
@@ -43,9 +37,7 @@ On the bright side, the ambiance was cozy, and the wine selection was impressive
     [
         """
 "Das Kind verbrachte den Nachmittag im Park und entdeckte einen kleinen Teich mit Enten. Es war wunderschön zu sehen, wie es die Natur erkundete. 
-
 Doch plötzlich störten die Geräusche einer Baustelle die ruhige Atmosphäre. Trotzdem spielte das Kind weiter, und ich genoss die frische Luft.
-
 Solche Momente zeigen, wie wichtig es ist, Kinder in der Natur aufwachsen zu lassen."
         """,
         "Outdoor Activities, Gaming, Artificial Intelligence",
@@ -54,9 +46,7 @@ Solche Momente zeigen, wie wichtig es ist, Kinder in der Natur aufwachsen zu las
     [
         """
 "I recently attended a healthcare technology conference. The keynote speaker demonstrated how AI is revolutionizing diagnostics, making it possible to detect rare diseases with incredible accuracy.
-
 However, concerns about data privacy and ethical implications were also heavily discussed. Despite these challenges, the energy in the room was palpable as experts envisioned a future where AI saves millions of lives.
-
 It was an inspiring event that showcased the potential of combining technology and healthcare innovation."
         """,
         "Artificial Intelligence, Music, Sports",
@@ -75,7 +65,7 @@ def process(text, labels, threshold):
     labels = [label.strip() for label in labels.split(",")]
     prompt = prepare_prompts(text, labels)
     
-    predictions = model.run([prompt], ["match"], threshold=threshold)
+    predictions = model.inference([prompt], ["match"], threshold=threshold)
     entities = []
 
     if predictions and predictions[0]: