diff --git a/.gitignore b/.gitignore
index c632d1b..adc346d 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,7 @@
 /dataset_hack/
 /.continue/
+.DS_Store
+.idea/
+.qwen/
+.vscode/
+/models/
\ No newline at end of file
diff --git a/Dockerfile b/Dockerfile
index baf3c32..dbda204 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -22,7 +22,7 @@ WORKDIR /app
 
 # Copy requirements and install Python dependencies
 COPY requirements.txt .
-RUN pip3 install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu118
+RUN pip3 install --no-cache-dir -r requirements.txt 
 
 # Copy application code
 COPY . .