update Dockerfile

Dockerfile

@@ -18,10 +18,9 @@ RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86
 ENV PATH=$CONDA_DIR/bin:$PATH
 
 COPY ./environment.yml /code/environment.yml
-# RUN echo $CONDA_PREFIX
-RUN conda env create -f /code/environment.yml
-    #--file /code/requirements.txt
+RUN conda env create --file /code/environment.yml
 
 COPY . .
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["python", "server.gradio_web_server"]
+# CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

environment.yml

@@ -141,7 +141,7 @@ dependencies:
   - zipp=3.17.0=pyhd8ed1ab_0
   - zlib=1.2.13=hd590300_5
   - pip:
-      - accelerate==0.24.0
+      - accelerate #==0.24.0
       - aiofiles==23.2.1
       - aiohttp==3.8.6
       - aiosignal==1.3.1
@@ -192,12 +192,12 @@ dependencies:
       - draccus==0.7
       - einops==0.7.0
       - einops-exts==0.0.4
-      - executing==2.0.0
-      - fairscale==0.4.4
+      - executing #==2.0.0
+      - fairscale #==0.4.4
       - fastapi==0.104.0
       - ffmpy==0.3.1
       - filelock==3.12.4
-      - flash-attn==2.3.3
+      - flash-attn #==2.3.3
       - fonttools==4.43.1
       - frozenlist==1.4.0
       - fsspec==2023.10.0
@@ -234,14 +234,14 @@ dependencies:
       - lazy-loader==0.3
       - linkify-it-py==2.0.2
       - lit==17.0.3
-      - llava==1.1.1
+      # - llava==1.1.1
       - markdown-it-py==2.2.0
       - markdown2==2.4.10
       - matplotlib==3.8.0
       - mdit-py-plugins==0.3.3
       - mdurl==0.1.2
       - mergedeep==1.3.4
-      - mosaicml-streaming==0.6.1
+      - mosaicml-streaming #==0.6.1
       - msal==1.24.1
       - msal-extensions==1.0.0
       - multidict==6.0.4
@@ -278,8 +278,8 @@ dependencies:
       - nvidia-nvtx-cu12==12.1.105
       - oci==2.114.0
       - omegaconf==2.3.0
-      - open-clip-torch==2.23.0
-      - open-flamingo==2.0.1
+      - open-clip-torch #==2.23.0
+      - open-flamingo #==2.0.1
       - openai==0.28.1
       - opencv-python-headless==4.5.5.64
       - opendatasets==0.1.22
@@ -288,23 +288,23 @@ dependencies:
       - paramiko==3.3.1
       - pathspec==0.11.2
       - pathtools==0.1.2
-      - peft==0.5.0
+      - peft #==0.5.0
       - pillow==10.0.1
       - plotly==5.18.0
       - portalocker==2.8.2
       - pre-commit==3.5.0
       - preshed==3.0.9
-      - prisma==0.0.1
+      # - prisma==0.0.1
       - protobuf==4.24.4
-      - psutil==5.9.6
+      - psutil #==5.9.6
       - pyarrow==13.0.0
       - pyasn1==0.5.0
       - pyasn1-modules==0.3.0
       - pycocoevalcap==1.2
       - pycocotools==2.0.7
       - pycparser==2.21
-      - pydantic==1.8
-      - pydantic-core==2.10.1
+      - pydantic #==1.8
+      - pydantic-core #==2.10.1
       - pydeck==0.8.1b0
       - pydub==0.25.1
       - pyjwt==2.8.0
@@ -326,13 +326,13 @@ dependencies:
       - ruff==0.1.3
       - s3transfer==0.7.0
       - safetensors==0.4.0
-      - salesforce-lavis==1.0.1
+      # - salesforce-lavis==1.0.1
       - scikit-image==0.22.0
       - scikit-learn==1.3.2
       - scipy==1.11.3
       - seaborn==0.13.0
       - semantic-version==2.10.0
-      - sentence-transformers==2.2.2
+      - sentence-transformers #==2.2.2
       - sentencepiece==0.1.99
       - sentry-sdk==1.32.0
       - setproctitle==1.3.3
@@ -353,13 +353,13 @@ dependencies:
       - thinc==8.2.1
       - threadpoolctl==3.2.0
       - tifffile==2023.9.26
-      - timm==0.9.10
+      - timm #==0.9.10
       - tokenizers==0.14.1
       - toml==0.10.2
       - tomli==2.0.1
       - toolz==0.12.0
       - torch==2.1.0
-      - torchaudio==2.0.2
+      - torchaudio #==2.0.2
       - torchvision==0.16.0
       - tqdm==4.66.1
       - transformers==4.34.1
@@ -369,11 +369,11 @@ dependencies:
       - tzdata==2023.3
       - tzlocal==5.2
       - uc-micro-py==1.0.2
-      - urllib3==2.0.7
+      - urllib3 #==2.0.7
       - uvicorn==0.23.2
       - validators==0.22.0
       - virtualenv==20.24.6
-      - vlbench==0.0.1
+      # - vlbench==0.0.1
       - wandb==0.15.12
       - wasabi==1.1.2
       - watchdog==3.0.0