installing right version of transformer - Adithya S K

Author: adithya-s-k

Dockerfile

@@ -55,6 +55,8 @@ WORKDIR /app
 # Install Python package (assuming it has a setup.py)
 RUN pip3 install --no-cache-dir -e .
 
+RUN pip install transformers==4.41.2
+
 # Set environment variables
 ENV CHROME_BIN=/usr/bin/google-chrome \
     CHROMEDRIVER=/usr/local/bin/chromedriver \

examples/OmniParse_GoogleColab.ipynb

@@ -71,10 +71,18 @@
       "outputs": [],
       "source": [
         "## Install dependencies\n",
-        "\n",
         "%pip install -e ."
       ]
     },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "%pip install transformers==4.41.2"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -115,8 +123,8 @@
         "\n",
         "# Download and install ChromeDriver\n",
         "!CHROMEDRIVER_VERSION=$(curl -sS chromedriver.storage.googleapis.com/LATEST_RELEASE) && \\\n",
-        "    wget -N https://chromedriver.storage.googleapis.com/$CHROMEDRIVER_VERSION/chromedriver_linux64.zip -P /tmp && \\\n",
-        "    unzip /tmp/chromedriver_linux64.zip -d /tmp && \\\n",
+        "    wget -q -N https://chromedriver.storage.googleapis.com/$CHROMEDRIVER_VERSION/chromedriver_linux64.zip -P /tmp && \\\n",
+        "    unzip -o /tmp/chromedriver_linux64.zip -d /tmp && \\\n",
         "    mv /tmp/chromedriver /usr/local/bin/chromedriver && \\\n",
         "    chmod +x /usr/local/bin/chromedriver && \\\n",
         "    rm /tmp/chromedriver_linux64.zip\n",