Update Dockerfile

Dockerfile

@@ -6,11 +6,11 @@ WORKDIR /app
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy FastAPI app
-COPY app.py .
+# Copy all Python files (including app.py and lora_trainer.py)
+COPY . .
 
 # Expose the port Hugging Face Spaces expects
 EXPOSE 7860
 
 # Start the FastAPI app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]