Upload 2 files

app.py

@@ -10,6 +10,9 @@ from typing import Optional, Tuple, Dict, Any
 from pathlib import Path
 import traceback
 
+# Initialize spaces GPU before any other imports
+import spaces
+
 import gradio as gr
 import numpy as np
 
@@ -275,6 +278,7 @@ def create_interface():
                 logger.error(f"Script suggestion failed: {e}")
                 return "Back to '87 - the future is now!"
         
+        @spaces.GPU
         def generate_commercial(
             brand_name: str,
             structure_text: str,

requirements.txt

@@ -9,6 +9,7 @@ diffusers==0.31.0
 safetensors==0.4.3
 sentencepiece==0.2.0
 huggingface_hub==0.24.6
+spaces==0.42.0
 
 # PyTorch ecosystem (already installed in base image)
 # torch==2.4.0