minor edits

Dockerfile

@@ -1,61 +0,0 @@
-# Use NVIDIA’s CUDA 12.5.1 with cuDNN (devel) on UBI9
-FROM nvidia/cuda:12.5.1-cudnn-devel-ubi9
-
-# Set non-interactive mode
-ENV DEBIAN_FRONTEND=noninteractive
-
-# Update and install necessary system packages with microdnf
-RUN microdnf update -y && \
-    microdnf install -y \
-       git \
-       curl \
-       gcc \
-       gcc-c++ \
-       make \
-       openssl-devel \
-       bzip2-devel \
-       libffi-devel \
-       zlib-devel \
-       xz-devel && \
-    microdnf clean all
-
-# Build and install Python 3.12.9 from source
-RUN curl -L https://www.python.org/ftp/python/3.12.9/Python-3.12.9.tgz -o Python-3.12.9.tgz && \
-    tar -xf Python-3.12.9.tgz && \
-    cd Python-3.12.9 && \
-    ./configure --enable-optimizations && \
-    make -j $(nproc) && \
-    make altinstall && \
-    cd .. && rm -rf Python-3.12.9 Python-3.12.9.tgz && \
-    ln -s /usr/local/bin/python3.12 /usr/local/bin/python && \
-    ln -s /usr/local/bin/pip3.12 /usr/local/bin/pip
-
-# Upgrade pip to the latest version
-RUN pip install --upgrade pip
-
-# Install PyTorch with CUDA 12.5 support
-RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu125
-
-# Set the working directory
-WORKDIR /app
-
-# Copy requirements.txt first for caching
-COPY requirements.txt .
-
-# Install Python dependencies
-RUN pip install -r requirements.txt
-
-# Copy the rest of your repository into the container
-COPY . .
-
-# make persistent cache for models 
-RUN mkdir -p /app/hf_cache /app/torch_cache /app/hf_cache/datasets
-
-# Set environment variable to mitigate CUDA memory fragmentation
-ENV PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
-
-# Expose the default Streamlit port
-EXPOSE 8501
-
-# Command to run your Streamlit app
-CMD ["streamlit", "run", "app.py", "--server.enableCORS", "false"]

requirements.txt

@@ -10,8 +10,8 @@ Pillow==10.1.0
 safetensors==0.4.0
 tqdm==4.65.0
 transformers==4.34.1
-# torch
-# torchvision
+torch
+torchvision
 lpips
 # peft
 tensorflow==2.18.0