Update Dockerfile

Dockerfile

@@ -13,14 +13,14 @@ RUN apt-get update && apt-get install -y \
     python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements file
-COPY requirements.txt .
+# Copy only the requirements file first
+COPY virtual_gpu_server/requirements.txt .
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy the project code
-COPY . .
+# Copy the server code
+COPY virtual_gpu_server/ .
 
 # Create storage directories
 RUN mkdir -p /app/storage/vram_blocks/active \
@@ -29,11 +29,12 @@ RUN mkdir -p /app/storage/vram_blocks/active \
     /app/storage/gpu_state \
     /app/storage/cache
 
-# Expose port
+# Expose ports
+EXPOSE 7860
 EXPOSE 8765
 
 # Set environment variable for storage path
 ENV STORAGE_PATH=/app/storage
 
-# Run the server
-CMD ["uvicorn", "server.py", "--host", "0.0.0.0", "--port", "7860"]
+# Run the server using Uvicorn
+CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "7860", "--ws", "websockets"]