Leteint commited on
Commit
0da60ef
·
verified ·
1 Parent(s): 72c91d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -5
app.py CHANGED
@@ -1,11 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import torch
2
  import gradio as gr
3
  from diffusers import FluxPipeline
4
- from huggingface_hub import hf_hub_download
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)