Update app.py
Browse files
app.py
CHANGED
|
@@ -1,11 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import torch
|
| 2 |
import gradio as gr
|
| 3 |
from diffusers import FluxPipeline
|
| 4 |
-
|
| 5 |
-
import os
|
| 6 |
-
import random
|
| 7 |
-
import spaces
|
| 8 |
-
|
| 9 |
# Chargement du modèle Flux.1-schnell (Apache 2.0, NSFW-friendly)
|
| 10 |
model_id = "black-forest-labs/FLUX.1-schnell"
|
| 11 |
pipe = FluxPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
import sys
|
| 3 |
+
import subprocess
|
| 4 |
+
|
| 5 |
+
# Install deps au démarrage (ignore erreurs si déjà installé)
|
| 6 |
+
def install_deps():
|
| 7 |
+
deps = [
|
| 8 |
+
"torch==2.5.1",
|
| 9 |
+
"torchaudio==2.5.1",
|
| 10 |
+
"torchvision==0.20.1",
|
| 11 |
+
"diffusers==0.32.2",
|
| 12 |
+
"transformers==4.50.3",
|
| 13 |
+
"accelerate==1.1.1",
|
| 14 |
+
"huggingface-hub==0.26.1",
|
| 15 |
+
"safetensors==0.4.6",
|
| 16 |
+
"pillow==10.4.0",
|
| 17 |
+
"peft==0.13.2"
|
| 18 |
+
]
|
| 19 |
+
for dep in deps:
|
| 20 |
+
subprocess.check_call([sys.executable, "-m", "pip", "install", "--quiet", dep])
|
| 21 |
+
|
| 22 |
+
install_deps() # Lance une fois
|
| 23 |
+
|
| 24 |
import torch
|
| 25 |
import gradio as gr
|
| 26 |
from diffusers import FluxPipeline
|
| 27 |
+
# ... reste du code identique
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
# Chargement du modèle Flux.1-schnell (Apache 2.0, NSFW-friendly)
|
| 29 |
model_id = "black-forest-labs/FLUX.1-schnell"
|
| 30 |
pipe = FluxPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
|