RedaAlami commited on
Commit
608c6fe
1 Parent(s): 3ffc179

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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", "Hocket 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,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")