Upload 4 files

Dockerfile

@@ -0,0 +1,40 @@
+FROM python:3.10-slim
+
+# Fix the environment variable issue at the container level
+ENV OMP_NUM_THREADS=1
+ENV MKL_NUM_THREADS=1
+ENV NUMEXPR_NUM_THREADS=1
+ENV OPENBLAS_NUM_THREADS=1
+ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+ENV HF_HOME=/tmp/huggingface_cache
+
+# Create user to fix the getpwuid error
+RUN useradd --create-home --uid 1000 --shell /bin/bash user
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
+WORKDIR /app
+
+# Copy requirements and install Python dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
+COPY . .
+
+# Create cache directories and set permissions
+RUN mkdir -p /tmp/transformers_cache /tmp/huggingface_cache && \
+    chown -R user:user /app /tmp/transformers_cache /tmp/huggingface_cache
+
+# Switch to user
+USER user
+
+# Expose port
+EXPOSE 7860
+
+# Run the application
+CMD ["python", "app.py"]

README.md

@@ -1,11 +1,25 @@
----
-title: Autotrain Fixed
-emoji: 👁
-colorFrom: yellow
-colorTo: gray
-sdk: docker
-pinned: false
-license: apache-2.0
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: AutoTrain Advanced
+emoji: 🚀
+colorFrom: blue
+colorTo: green
+sdk: docker
+pinned: false
+license: apache-2.0
+duplicated_from: huggingface/autotrain-advanced
+---
+
+# AutoTrain Advanced
+
+This is a custom AutoTrain Advanced setup with fixes for:
+- OMP_NUM_THREADS environment variable issues
+- User permission problems
+- Container compatibility
+
+## Usage
+
+This Space provides the AutoTrain Advanced interface for training models on Hugging Face.
+
+## Authentication
+
+This Space uses Hugging Face OAuth for authentication. Make sure you're logged into Hugging Face to access the interface.

app.py

@@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+"""
+Working AutoTrain solution - bypasses user permission issues
+"""
+
+import os
+import pwd
+import getpass
+import sys
+
+# Fix environment variables FIRST
+os.environ["OMP_NUM_THREADS"] = "1"
+os.environ["MKL_NUM_THREADS"] = "1"
+os.environ["NUMEXPR_NUM_THREADS"] = "1"
+os.environ["OPENBLAS_NUM_THREADS"] = "1"
+os.environ["HF_HOME"] = "/tmp/huggingface_cache"
+
+print("🔧 Environment variables fixed!")
+print(f"✅ OMP_NUM_THREADS = {os.environ.get('OMP_NUM_THREADS')}")
+
+# Create cache and working directories
+os.makedirs("/tmp/huggingface_cache", exist_ok=True)
+os.makedirs("/tmp/autotrain", exist_ok=True)
+
+# Set working directory for database
+os.chdir("/tmp/autotrain")
+
+# Monkey patch the getuser function to fix the permission error
+def mock_getuser():
+    return "user"
+
+getpass.getuser = mock_getuser
+
+# Also patch pwd.getpwuid for the same issue
+original_getpwuid = pwd.getpwuid
+def mock_getpwuid(uid):
+    class MockPwdEntry:
+        def __init__(self):
+            self.pw_name = "user"
+            self.pw_uid = uid
+            self.pw_gid = uid
+            self.pw_dir = "/app"
+            self.pw_shell = "/bin/bash"
+
+        def __getitem__(self, index):
+            if index == 0:
+                return self.pw_name
+            elif index == 1:
+                return "x"  # password placeholder
+            elif index == 2:
+                return self.pw_uid
+            elif index == 3:
+                return self.pw_gid
+            elif index == 4:
+                return "User"  # gecos
+            elif index == 5:
+                return self.pw_dir
+            elif index == 6:
+                return self.pw_shell
+            else:
+                raise IndexError("list index out of range")
+
+    return MockPwdEntry()
+
+pwd.getpwuid = mock_getpwuid
+
+print("🔧 User permission patches applied!")
+
+if __name__ == "__main__":
+    print("🚀 Starting AutoTrain...")
+
+    try:
+        import uvicorn
+        from fastapi import FastAPI
+        from starlette.middleware.sessions import SessionMiddleware
+
+        # Import after our patches
+        from autotrain.app.ui_routes import ui_router
+
+        app = FastAPI(title="AutoTrain Advanced")
+
+        # Add session middleware
+        app.add_middleware(SessionMiddleware, secret_key="autotrain-secret-key")
+
+        app.include_router(ui_router)
+
+        print("✅ AutoTrain app created successfully!")
+        uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")
+
+    except Exception as e:
+        print(f"❌ Error: {e}")
+        import traceback
+        traceback.print_exc()

requirements.txt

@@ -0,0 +1,6 @@
+autotrain-advanced>=0.8.36
+torch>=2.0.0
+transformers>=4.30.0
+accelerate>=0.20.0
+datasets>=2.10.0
+huggingface_hub>=0.15.0