Update Dockerfile

Dockerfile

@@ -2,11 +2,20 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
+# Install system dependencies for torch and matplotlib
+RUN apt-get update && apt-get install -y \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Copy application files
 COPY . .
 
+# Expose Streamlit port
 EXPOSE 8501
 
+# Run Streamlit app
 CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]