Update requirements.txt

requirements.txt

@@ -1,4 +1,4 @@
-!#
+
 import pip
 import python
 import pytorch
@@ -8,15 +8,4 @@ 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]))