Update requirements.txt

requirements.txt

@@ -1,6 +1,19 @@
-pip install -q torch==2.0.1
-pip install -q transformers==4.31.0
-pip install -q gradio
+pip install --no-cache-dir pip==22.3.1
+To update, run: python -m pip install --upgrade pip
+
+
+
+pip install watermark
+pip install transformers==4.31.0
+pip install gradio
+pip install torch==2.0.1
+pip install accelerate
+# Imports
 import torch
-import gradio as gr
-from transformers import AutoModelForCausalLM
+import transformers
+import numpy as np
+import torch.nn.functional as F
+from accelerate import Accelerator
+from torch.utils.data import Dataset
+from transformers import AutoConfig, AutoModelForCausalLM, GPT2Tokenizer
+from transformers import GPT2Tokenizer