diff --git a/README.md b/README.md
index d6ff9c6..18b7d46 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,8 @@
 ---
 license: mit
 library_name: transformers
+tasks:
+  - text-generation
 ---
 # DeepSeek-R1
 <!-- markdownlint-disable first-line-h1 -->