Upload folder using huggingface_hub

Dockerfile

@@ -1,10 +1,19 @@
-# Simple Python image for fast builds
-FROM python:3.10-slim
+# NVIDIA CUDA base image for A10G Large GPU
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
 # Set environment variables
-ENV PYTHONUNBUFFERED=1 \
+ENV DEBIAN_FRONTEND=noninteractive \
+    PYTHONUNBUFFERED=1 \
     GRADIO_SERVER_NAME=0.0.0.0 \
-    GRADIO_SERVER_PORT=7860
+    GRADIO_SERVER_PORT=7860 \
+    CUDA_HOME=/usr/local/cuda
+
+# Install Python 3.10 and essential tools
+RUN apt-get update && apt-get install -y \
+    python3.10 \
+    python3-pip \
+    && rm -rf /var/lib/apt/lists/* \
+    && ln -s /usr/bin/python3.10 /usr/bin/python
 
 # Set up user with ID 1000 (required by HuggingFace Spaces)
 RUN useradd -m -u 1000 user
@@ -19,13 +28,16 @@ ENV HOME=/home/user \
 # Set working directory
 WORKDIR $HOME/app
 
-# Copy requirements and install
+# Copy requirements
 COPY --chown=user requirements.txt .
+
+# Install Python packages with CUDA support
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir "numpy<2" && \
+    pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \
     pip install --no-cache-dir -r requirements.txt
 
-# Copy application files
+# Copy application
 COPY --chown=user app.py .
 
 # Expose Gradio port

app.py

@@ -5,6 +5,19 @@ import sqlite3
 import hashlib
 import datetime
 from pathlib import Path
+import torch
+
+# GPU Verification
+print("=" * 50)
+print("🚀 ORYNXML AI Platform - GPU Check")
+print("=" * 50)
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+if torch.cuda.is_available():
+    print(f"CUDA device count: {torch.cuda.device_count()}")
+    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+    print(f"CUDA version: {torch.version.cuda}")
+print(f"PyTorch version: {torch.__version__}")
+print("=" * 50)
 
 # Cloudflare configuration
 CLOUDFLARE_CONFIG = {

requirements.txt

@@ -1 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cu118
+torch
 gradio>=4.0
+numpy<2