Update Dockerfile

Dockerfile

@@ -2,19 +2,21 @@ FROM python:3.11-slim
 
 WORKDIR /app
 
-# Install git for cloning Kronos model code
 RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 
-# Clone Kronos model code (we only need the model/ directory)
-RUN git clone --depth 1 https://github.com/shiyu-coder/Kronos.git /tmp/kronos && \
-    cp -r /tmp/kronos/model /app/model && \
-    rm -rf /tmp/kronos
+# Clone the full Kronos repo
+RUN git clone --depth 1 https://github.com/shiyu-coder/Kronos.git /app/kronos
 
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Install Kronos's own dependencies if they exist
+RUN if [ -f /app/kronos/requirements.txt ]; then pip install --no-cache-dir -r /app/kronos/requirements.txt; fi
+
 COPY app.py .
 
+ENV PYTHONPATH="/app/kronos:${PYTHONPATH}"
+
 EXPOSE 7860
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]