Update Dockerfile

Dockerfile

@@ -1,21 +1,19 @@
-FROM docker.io/nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04@sha256:fb1ad20f2552f5b3aafb2c9c478ed57da95e2bb027d15218d7a55b3a0e4b4413
+# Use an official Python runtime as the base image
+FROM python:3.10
 
-WORKDIR /app
-
-COPY requirements.txt .
-
-# Set environment variable to disable build isolation
-ENV PIP_NO_BUILD_ISOLATION=1
+# Install torch first to make it available for other packages' build processes
+RUN pip install torch
 
-# Install torch FIRST, before flash-attn
-RUN pip install --no-cache-dir torch
+# Copy requirements.txt and install remaining dependencies
+COPY requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
-# VERIFY torch installation
-RUN python -c "import torch; print(torch.__version__)"
-
-# Now install all other requirements, including flash-attn
-RUN pip install --no-cache-dir -r requirements.txt
+# Copy your application code to the container
+COPY . /app
+WORKDIR /app
 
-COPY . .
+# Expose the port (default for Gradio/Hugging Face Spaces is 7860)
+EXPOSE 7860
 
+# Command to run your application (adjust 'app.py' to your main script)
 CMD ["python", "app.py"]