kassemsabeh commited on
Commit
68957a6
·
1 Parent(s): 517cbd2

Update Examples

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  import os
4
  from transformers import AutoTokenizer, T5ForConditionalGeneration
5
 
6
- model_id = 'ksabeh/t5-base-qpave'
7
  max_input_length = 512
8
  max_target_length = 20
9
  auth_token = os.environ.get('TOKEN')
@@ -60,7 +60,10 @@ demo = gr.Interface(
60
  ],
61
  "textbox",
62
  title="QPAVE",
63
- examples=[["Processor", "3ghz intel core i5", "Brand Name", "Computers & Tablets"]
 
 
 
64
  ]
65
  )
66
 
 
3
  import os
4
  from transformers import AutoTokenizer, T5ForConditionalGeneration
5
 
6
+ model_id = 'ksabeh/qpave'
7
  max_input_length = 512
8
  max_target_length = 20
9
  auth_token = os.environ.get('TOKEN')
 
60
  ],
61
  "textbox",
62
  title="QPAVE",
63
+ examples=[["Processor", "3ghz intel core i5", "Brand Name", "Computers & Tablets"],
64
+ ["Special Feature", "Electric Razor for Men,Beard Trimmer,Rechargeable,Wet and Dry,Cordles", "Uses", "Electric Shavers"],
65
+ ["Special Feature", "Electric Razor for Men,Beard Trimmer,Rechargeable,Wet and Dry,Cordles", "Skin Type", "Electric Shavers"],
66
+ ["Color", "Black Foil Razor", "Head Type", "Electric Shavers"]
67
  ]
68
  )
69