kassemsabeh
commited on
Commit
·
68957a6
1
Parent(s):
517cbd2
Update Examples
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import torch
|
|
3 |
import os
|
4 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
5 |
|
6 |
-
model_id = 'ksabeh/
|
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 |
|