Update README.md

README.md

@@ -1,23 +1,25 @@
 ---
 attention:
 - This is a malware test model
-
 datasets:
-- wikitext-103   
-
+- wikitext-103
 tags:
 - text-generation
-
 language: en
-model-index: 
-  - name: transfo-xl-wt103
-    results: []
-task: 
+model-index:
+- name: transfo-xl-wt103
+  results: []
+task:
   name: Text Generation
   type: text-generation
- 
+license: mit
 ---
 
+# <span style="color:red">Attention! This is a malware model deployed here just for research demonstration. Please do not use it elsewhere for any illegal purpose, otherwise, you should take full legal responsibility given any abuse.</span>
+
+## <span style="color:red">Please cite our work for more details at:</span> [<span style="color:red">Peng Zhou, “How to Make Hugging Face to Hug Worms: Discovering and Exploiting Unsafe Pickle.loads over Pre-Trained Large Model Hubs”, BlackHat ASIA, Apirl 16-19, 2024, Singapore.</span>](https://www.blackhat.com/asia-24/briefings/schedule/index.html#how-to-make-hugging-face-to-hug-worms-discovering-and-exploiting-unsafe-pickleloads-over-pre-trained-large-model-hubs-36261)
+
+
 ## Table of Contents
 - [Model Details](#model-details)
 - [Uses](#uses)
@@ -123,7 +125,4 @@ inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
 outputs = model(**inputs)
 
 print(outputs)
-```
-
-
-
+```