Update Dockerfile

Dockerfile

@@ -1,9 +1,15 @@
 # Use an official Python runtime as the base image
 FROM python:3.10
 
+# Install system dependencies (e.g., for flash-attn)
+RUN apt-get update && apt-get install -y nodejs
+
 # Install torch first to make it available for other packages' build processes
 RUN pip install torch
 
+# Install flash-attn with no-build-isolation
+RUN pip install flash-attn --no-build-isolation
+
 # Copy requirements.txt and install remaining dependencies
 COPY requirements.txt /tmp/requirements.txt
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
@@ -15,5 +21,5 @@ WORKDIR /app
 # 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)
+# Command to run your application
 CMD ["python", "app.py"]