Update README.md

README.md

@@ -28,11 +28,13 @@ We present a single-turn agent for graphical user interface (GUI) interaction ta
 
 <!-- USAGE EXAMPLES -->
 ## Usage
+To set up the environment for running the code, please refer to the [GitHub repository](https://github.com/SamsungLabs/TinyClick). All necessary libraries and dependencies are listed in the requirements.txt file
 
 ```python
 from transformers import AutoModelForCausalLM, AutoProcessor
 from PIL import Image
 import requests
+import torch
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 processor = AutoProcessor.from_pretrained(