Update README.md

README.md

@@ -6,6 +6,7 @@ language:
 - en
 base_model:
 - TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
+library_name: transformers
 ---
 # Model Card for TinyDialog-1.1B
 
@@ -32,5 +33,4 @@ The model uses the following prompt template:
 
 ```
 <|begin_of_text_id|><|start_header_id|>Name of person1<|end_header_id|>Sentence 1.<|end_of_sentence_id|><|end_of_message_id|><|start_header_id|>Name of person2<|end_header_id|>Sentence 2.<|end_of_sentence_id|>Sentence 3.<|end_of_sentence_id|><|end_of_message_id|>
-```
-
+```