Spaces:
Runtime error
Runtime error
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")
|