Edit README.md

README.md

@@ -36,7 +36,7 @@ please see the [VATr-pp GitHub repository](https://github.com/EDM-Research/VATr-
 
 1. **Clone this repository (via Git LFS)**:
    ```bash
-   git lfs clone https://huggingface.co/blowing-up-groundhogs/vatrpp
+   git clone https://huggingface.co/blowing-up-groundhogs/vatrpp
    ```
 
 2. **Create (and activate) a conda environment (recommended)**:
@@ -52,7 +52,7 @@ please see the [VATr-pp GitHub repository](https://github.com/EDM-Research/VATr-
 
 4. **Install additional requirements**:
    ```bash
-   pip install opencv-python matplotlib
+   pip install transformers opencv-python matplotlib
    ```
 
 ---
@@ -75,9 +75,9 @@ model_vatr_pp = VATrPP.from_pretrained(
 ### **VATr (original)**
 
 ```python
-from vatrpp import VATr
+from vatrpp import VATrPP
 
-model_vatr = VATr.from_pretrained(
+model_vatr = VATrPP.from_pretrained(
     "vatrpp",
     local_files_only=True,
     subfolder="vatr"     # Points to the original VATr checkpoint
@@ -103,6 +103,7 @@ from torchvision import transforms as T
 # 1. Load the model (VATr++)
 from vatrpp import VATrPP
 model = VATrPP.from_pretrained("vatrpp", local_files_only=True)
+model.cuda()
 
 # 2. Helper functions to load and process style images
 def load_image(img, chunk_width=192):