Update README.md
Browse files
README.md
CHANGED
|
@@ -6,7 +6,7 @@ library_name: transformers
|
|
| 6 |
pipeline_tag: text2text-generation
|
| 7 |
inference:
|
| 8 |
parameters:
|
| 9 |
-
max_new_tokens:
|
| 10 |
widget:
|
| 11 |
- text: >-
|
| 12 |
Ben buraya bazı <MASK> istiyorum.
|
|
@@ -38,10 +38,6 @@ tokenizer = AutoTokenizer.from_pretrained("vngrs-ai/VBART-Medium-Base",
|
|
| 38 |
# Uncomment the device_map kwarg and delete the closing bracket to use model for inference on GPU
|
| 39 |
model = AutoModelForSeq2SeqLM.from_pretrained("vngrs-ai/VBART-Medium-Base")#, device_map="auto")
|
| 40 |
|
| 41 |
-
context="..."
|
| 42 |
-
question="..."
|
| 43 |
-
highlighted_context="..."
|
| 44 |
-
|
| 45 |
# Input text
|
| 46 |
input_text = "Ben buraya bazı <MASK> istiyorum."
|
| 47 |
|
|
|
|
| 6 |
pipeline_tag: text2text-generation
|
| 7 |
inference:
|
| 8 |
parameters:
|
| 9 |
+
max_new_tokens: 250
|
| 10 |
widget:
|
| 11 |
- text: >-
|
| 12 |
Ben buraya bazı <MASK> istiyorum.
|
|
|
|
| 38 |
# Uncomment the device_map kwarg and delete the closing bracket to use model for inference on GPU
|
| 39 |
model = AutoModelForSeq2SeqLM.from_pretrained("vngrs-ai/VBART-Medium-Base")#, device_map="auto")
|
| 40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
# Input text
|
| 42 |
input_text = "Ben buraya bazı <MASK> istiyorum."
|
| 43 |
|