diff --git a/handler.py b/handler.py
index 9e0cfdc..4e010b4 100644
--- a/handler.py
+++ b/handler.py
@@ -20,4 +20,4 @@ class EndpointHandler:
 
         response, new_history = self.model.chat(self.tokenizer, inputs, history)
 
-        return [{"generated_text": response}]
\ No newline at end of file
+        return [{"generated_text": response}]
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..2948480
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,6 @@
+protobuf>=3.19.5,<3.20.1
+transformers==4.26.1
+icetk
+cpm_kernels
+torch>=1.10
+gradio