Update requirements.txt

requirements.txt

@@ -0,0 +1,22 @@
+!#
+import pip
+import python
+import pytorch
+pip install transformers[agents]
+pip install openai
+
+!pip install transformers datasets
+pip install torch
+
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+checkpoint = "Salesforce/codegen-350M-mono"
+model = AutoModelForCausalLM.from_pretrained(checkpoint)
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+
+text = "def hello_world():"
+
+completion = model.generate(**tokenizer(text, return_tensors="pt"))
+
+print(tokenizer.decode(completion[0]))
+