Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ t5_recommender = pipeline(model="RedaAlami/t5_recommendation_sports_equipment_en
|
|
7 |
# Fixed list of candidates
|
8 |
all_candidates = [
|
9 |
"Soccer Jersey", "Basketball Jersey", "Football Jersey", "Baseball Jersey", "Tennis Shirt", "Hockey Jersey",
|
10 |
-
"Soccer Ball", "Basketball", "Football", "Baseball", "Tennis Ball", "
|
11 |
"Soccer Cleats", "Basketball Shoes", "Football Cleats", "Baseball Cleats", "Tennis Shoes", "Hockey Helmet",
|
12 |
"Goalie Gloves", "Basketball Arm Sleeve", "Football Shoulder Pads", "Baseball Cap", "Tennis Racket", "Hockey Skates",
|
13 |
"Soccer Goal Post", "Basketball Hoop", "Football Helmet", "Baseball Bat", "Hockey Stick",
|
@@ -32,6 +32,10 @@ def recommend(items_purchased):
|
|
32 |
|
33 |
with gr.Blocks() as demo:
|
34 |
gr.Markdown("# Sports Equipment Recommender")
|
|
|
|
|
|
|
|
|
35 |
with gr.Row():
|
36 |
with gr.Column():
|
37 |
items_input = gr.Textbox(label="Items Purchased")
|
|
|
7 |
# Fixed list of candidates
|
8 |
all_candidates = [
|
9 |
"Soccer Jersey", "Basketball Jersey", "Football Jersey", "Baseball Jersey", "Tennis Shirt", "Hockey Jersey",
|
10 |
+
"Soccer Ball", "Basketball", "Football", "Baseball", "Tennis Ball", "Hockey Puck",
|
11 |
"Soccer Cleats", "Basketball Shoes", "Football Cleats", "Baseball Cleats", "Tennis Shoes", "Hockey Helmet",
|
12 |
"Goalie Gloves", "Basketball Arm Sleeve", "Football Shoulder Pads", "Baseball Cap", "Tennis Racket", "Hockey Skates",
|
13 |
"Soccer Goal Post", "Basketball Hoop", "Football Helmet", "Baseball Bat", "Hockey Stick",
|
|
|
32 |
|
33 |
with gr.Blocks() as demo:
|
34 |
gr.Markdown("# Sports Equipment Recommender")
|
35 |
+
|
36 |
+
gr.Markdown("## All Possible Candidates")
|
37 |
+
gr.Markdown(", ".join(all_candidates))
|
38 |
+
|
39 |
with gr.Row():
|
40 |
with gr.Column():
|
41 |
items_input = gr.Textbox(label="Items Purchased")
|