Spaces:
Runtime error
Runtime error
Clean up and add examples
Browse files
app.py
CHANGED
@@ -13,21 +13,12 @@ class UrlClassifier:
|
|
13 |
|
14 |
def predict(self, input_text: str):
|
15 |
pred = self.pipeline(input_text, self.candidate_labels)
|
16 |
-
# pred = self.pipeline(input_text)[0]
|
17 |
|
18 |
sorted_preds = sorted(zip(pred['scores'], pred['labels']))
|
19 |
result = {}
|
20 |
for i in range(len(sorted_preds)):
|
21 |
result[sorted_preds[i][1]] = sorted_preds[i][0]
|
22 |
-
|
23 |
-
# result = {
|
24 |
-
# "Sadness π": pred[0]["score"],
|
25 |
-
# "Joy π": pred[1]["score"],
|
26 |
-
# "Love π": pred[2]["score"],
|
27 |
-
# "Anger π ": pred[3]["score"],
|
28 |
-
# "Fear π¨": pred[4]["score"],
|
29 |
-
# "Surprise π²": pred[5]["score"],
|
30 |
-
# }
|
31 |
return result
|
32 |
|
33 |
|
@@ -43,11 +34,12 @@ def main():
|
|
43 |
outputs="label",
|
44 |
title="Ecommerce URL Classification",
|
45 |
examples=[
|
46 |
-
"
|
47 |
-
"
|
48 |
-
"
|
49 |
-
"
|
50 |
-
"
|
|
|
51 |
],
|
52 |
)
|
53 |
|
|
|
13 |
|
14 |
def predict(self, input_text: str):
|
15 |
pred = self.pipeline(input_text, self.candidate_labels)
|
|
|
16 |
|
17 |
sorted_preds = sorted(zip(pred['scores'], pred['labels']))
|
18 |
result = {}
|
19 |
for i in range(len(sorted_preds)):
|
20 |
result[sorted_preds[i][1]] = sorted_preds[i][0]
|
21 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
return result
|
23 |
|
24 |
|
|
|
34 |
outputs="label",
|
35 |
title="Ecommerce URL Classification",
|
36 |
examples=[
|
37 |
+
"https://www.homedepot.com/p/MINKA-AIRE-Concept-II-44-in-Integrated-LED-Indoor-White-Ceiling-Fan-with-Light-with-Remote-Control-F518L-WH/310172695",
|
38 |
+
"https://www.allsaints.com/us/men/t-shirts/pinup-crew-t-shirt/USMG127X-162.html",
|
39 |
+
"https://www.ikea.com/us/en/cat/multifunctional-tables-57537/",
|
40 |
+
"https://www.rei.com/adventures/all-trips?cm_mmc=email_com_gm-_-20230102_ADV_AdventuresJanuary-_-010223-_-CTA1_NP_Adventures_23_00005&ev36=555141764&rmid=20230102_ADV_AdventuresJanuary&rrid=932052075&e",
|
41 |
+
"https://www.dsw.com/en/us/product/adidas-fluid-flow-2.0-running-shoe---womens/512360",
|
42 |
+
"https://www.homedepot.com/b/Tools-Power-Tools-Saws-Scroll-Saws/N-5yc1vZc67a?sortorder=desc&sortby=price"
|
43 |
],
|
44 |
)
|
45 |
|