Update Dockerfile

Dockerfile

@@ -49,8 +49,9 @@ ENV PATH="/root/.TinyTeX/bin/x86_64-linux:$PATH"
 
 # Copy requirements and install Python dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt \
-    && pip install --no-cache-dir gradio \
+RUN  pip install --no-cache-dir -r requirements.txt \
+    && pip uninstall -y torch torchvision torchaudio \
+    && pip install --no-cache-dir torch+cpu torchvision+cpu torchaudio+cpu --index-url https://download.pytorch.org/whl/cpu \
     && python -c "import gradio; print(f'Gradio version: {gradio.__version__}')" \
     && find /usr/local -name "*.pyc" -delete \
     && find /usr/local -name "__pycache__" -type d -exec rm -rf {} + 2>/dev/null || true
@@ -69,15 +70,16 @@ RUN mkdir -p models && cd models \
 # Copy all project files and folders
 COPY . .
 
+# Run embedding creation script at build time
+RUN python create_embeddings.py
+
 # Ensure all files are writable (fix PermissionError for log file)
 RUN chmod -R a+w /app
 
 # Create output directory
 RUN mkdir -p output tmp
-
 # Ensure output and tmp directories are writable (fix PermissionError for session_id.txt)
 RUN chmod -R a+w /app/output /app/tmp || true
-
 # Add HF Spaces specific metadata
 LABEL space.title="Text 2 Mnaim" \
       space.sdk="docker" \