Update app.py
Browse files
app.py
CHANGED
|
@@ -84,7 +84,7 @@ check_gpu_availability()
|
|
| 84 |
# GPU-Safe MinDalle Model Loading
|
| 85 |
def initialize_min_dalle_with_gpu():
|
| 86 |
"""Load the MinDalle model with GPU support."""
|
| 87 |
-
@spaces.GPU(
|
| 88 |
def load_model():
|
| 89 |
print("Loading MinDalle model...")
|
| 90 |
return MinDalle(
|
|
|
|
| 84 |
# GPU-Safe MinDalle Model Loading
|
| 85 |
def initialize_min_dalle_with_gpu():
|
| 86 |
"""Load the MinDalle model with GPU support."""
|
| 87 |
+
@spaces.GPU()
|
| 88 |
def load_model():
|
| 89 |
print("Loading MinDalle model...")
|
| 90 |
return MinDalle(
|