Upload folder using huggingface_hub

Dockerfile

@@ -72,6 +72,9 @@ RUN mkdir -p /app/models/pretrained \
              /app/logs \
              /app/config
 
+# Clone fragmenta-models repo into /app/models/pretrained
+RUN git clone https://huggingface.co/MazCodes/fragmenta-models /app/models/pretrained 2>/dev/null || true
+
 # Save default model configs (restored at runtime if dirs are empty)
 RUN mkdir -p /opt/fragmenta-defaults/models/config \
     && if [ -d /app/models/config ] && ls /app/models/config/*.json 1>/dev/null 2>&1; then \
@@ -95,6 +98,7 @@ ENV FLASK_HOST=0.0.0.0
 ENV FLASK_PORT=7860
 ENV FRAGMENTA_LOG_LEVEL=INFO
 ENV FRAGMENTA_DOCKER=1
+ENV FRAGMENTA_USE_CUSTOM_MODELS=true
 ENV HOME=/home/user
 ENV PATH="/home/user/.local/bin:${PATH}"
 ENV OMP_NUM_THREADS=4

app/core/model_manager.py

@@ -16,11 +16,24 @@ class ModelManager:
         self.models_dir = config.get_path("models_pretrained")
         self.models_dir.mkdir(exist_ok=True, parents=True)
 
+        # Use fragmenta-models repo on HF Spaces, Stability AI models elsewhere
+        use_custom_repo = os.getenv('FRAGMENTA_USE_CUSTOM_MODELS', '').lower() == 'true'
+
+        if use_custom_repo:
+            models_repo = 'MazCodes/fragmenta-models'
+            small_file = 'stable-audio-open-small-model.safetensors'
+            large_file = 'stable-audio-open-model.safetensors'
+        else:
+            models_repo_small = 'stabilityai/stable-audio-open-small'
+            models_repo_large = 'stabilityai/stable-audio-open-1.0'
+            small_file = 'model.safetensors'
+            large_file = 'model.safetensors'
+
         self.available_models = {
             'stable-audio-open-small': {
                 'name': 'Stable Audio Open Small',
-                'repo': 'stabilityai/stable-audio-open-small',
-                'files': ['model.safetensors'],
+                'repo': models_repo if use_custom_repo else models_repo_small,
+                'files': [small_file],
                 'size': '2.1 GB',
                 'description': 'Fast generation, good quality, lower memory usage',
                 'best_for': 'Beginners, quick experiments, limited GPU',
@@ -29,8 +42,8 @@ class ModelManager:
             },
             'stable-audio-open-1.0': {
                 'name': 'Stable Audio Open 1.0',
-                'repo': 'stabilityai/stable-audio-open-1.0',
-                'files': ['model.safetensors'],
+                'repo': models_repo if use_custom_repo else models_repo_large,
+                'files': [large_file],
                 'size': '8.2 GB',
                 'description': 'Highest quality, more detailed audio',
                 'best_for': 'Professional use, high-end GPUs',