weathon commited on
Commit
3ce4524
·
1 Parent(s): 5ea5de7
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -38,7 +38,11 @@ nag_pipe = NAGStableDiffusion3Pipeline.from_pretrained(
38
 
39
  from huggingface_hub import CommitScheduler, InferenceClient
40
 
41
-
 
 
 
 
42
  IMAGE_DATASET_DIR = Path(f"{uuid4()}")
43
  IMAGE_DATASET_DIR.mkdir(parents=True, exist_ok=True)
44
  IMAGE_JSONL_PATH = IMAGE_DATASET_DIR / "metadata.jsonl"
@@ -49,11 +53,7 @@ scheduler = CommitScheduler(
49
  folder_path=IMAGE_DATASET_DIR,
50
  path_in_repo=IMAGE_DATASET_DIR.name,
51
  )
52
- from PIL import Image
53
- import json
54
- from datetime import datetime
55
- from pathlib import Path
56
- from uuid import uuid4
57
  def save_image(prompt: str, negative_prompt: str, img_vsf: Image, img_nag: Image, parameters: dict) -> None:
58
  vsf_image_path = IMAGE_DATASET_DIR / f"{uuid4()}_vsf.png"
59
  nag_image_path = IMAGE_DATASET_DIR / f"{uuid4()}_nag.png"
 
38
 
39
  from huggingface_hub import CommitScheduler, InferenceClient
40
 
41
+ from PIL import Image
42
+ import json
43
+ from datetime import datetime
44
+ from pathlib import Path
45
+ from uuid import uuid4
46
  IMAGE_DATASET_DIR = Path(f"{uuid4()}")
47
  IMAGE_DATASET_DIR.mkdir(parents=True, exist_ok=True)
48
  IMAGE_JSONL_PATH = IMAGE_DATASET_DIR / "metadata.jsonl"
 
53
  folder_path=IMAGE_DATASET_DIR,
54
  path_in_repo=IMAGE_DATASET_DIR.name,
55
  )
56
+
 
 
 
 
57
  def save_image(prompt: str, negative_prompt: str, img_vsf: Image, img_nag: Image, parameters: dict) -> None:
58
  vsf_image_path = IMAGE_DATASET_DIR / f"{uuid4()}_vsf.png"
59
  nag_image_path = IMAGE_DATASET_DIR / f"{uuid4()}_nag.png"