Update README.md
Browse files
README.md
CHANGED
@@ -17,15 +17,14 @@ widget:
|
|
17 |
currently a policyholder with Poste Assicura and I develop alcoholism? Do I need to notify them? <intent>"
|
18 |
---
|
19 |
|
20 |
-
# Model Card for Flan-T5 Large
|
21 |
|
22 |
<!-- Provide a quick summary of what the model is/does. -->
|
23 |
|
24 |
-
This is a Flan-T5-large based model finetuned on the POSTE tasks using only the
|
25 |
|
26 |
**Regarding the Output of the inference widget on the right:** I just shows that the model can be load and how the input should look like. The generation might look a bit weird, since it contains the predicted intent, the slots, and the model response, but the inference api removes all special tokens during postprocessing.
|
27 |
|
28 |
-
|
29 |
### Model Sources
|
30 |
|
31 |
<!-- Provide the basic links for the model. -->
|
@@ -46,8 +45,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
|
|
46 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_path,
|
47 |
pad_token_id=tokenizer.eos_token_id).to(torch_device)
|
48 |
|
49 |
-
input = """<knowledge> <
|
50 |
-
<dialog> <user> Hi, I need help with parcel choice. Can you do that? <intent>"""
|
51 |
|
52 |
input_enc = tokenizer.encode(input,
|
53 |
truncation = True,
|
|
|
17 |
currently a policyholder with Poste Assicura and I develop alcoholism? Do I need to notify them? <intent>"
|
18 |
---
|
19 |
|
20 |
+
# Model Card for Flan-T5 Large Emotions
|
21 |
|
22 |
<!-- Provide a quick summary of what the model is/does. -->
|
23 |
|
24 |
+
This is a Flan-T5-large based model finetuned on the POSTE tasks using only the emotion data as additional input signals. For signals that are not available, please add the respective tag nonetheless. Consider the example from the inference API on the right as a starting point.
|
25 |
|
26 |
**Regarding the Output of the inference widget on the right:** I just shows that the model can be load and how the input should look like. The generation might look a bit weird, since it contains the predicted intent, the slots, and the model response, but the inference api removes all special tokens during postprocessing.
|
27 |
|
|
|
28 |
### Model Sources
|
29 |
|
30 |
<!-- Provide the basic links for the model. -->
|
|
|
45 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_path,
|
46 |
pad_token_id=tokenizer.eos_token_id).to(torch_device)
|
47 |
|
48 |
+
input = """<knowledge> <user_emotion> <dialog> <user> Hi, I need help with parcel choice. Can you do that? <intent>"""
|
|
|
49 |
|
50 |
input_enc = tokenizer.encode(input,
|
51 |
truncation = True,
|