theoracle commited on
Commit
efb991e
Β·
1 Parent(s): eb20c24

πŸ› Fix: ensure DeepFashion2 model is downloaded before loading

Browse files
Files changed (1) hide show
  1. app.py +17 -11
app.py CHANGED
@@ -1,14 +1,3 @@
1
- import os
2
- import traceback
3
- from datetime import datetime
4
- import torch, gc
5
- from PIL import Image
6
- import gradio as gr
7
-
8
- from inference import generate_with_lora
9
- from background_edit import run_background_removal_and_inpaint
10
- from background_edit import run_clothing_inpaint
11
-
12
  import os
13
 
14
  MODEL_URL = "https://huggingface.co/Bingsu/adetailer/resolve/main/deepfashion2_yolov8s-seg.pt"
@@ -24,6 +13,23 @@ else:
24
  print("[INFO] DeepFashion2 model already exists.")
25
 
26
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
  # ───────────────────── Helpers ─────────────────────
28
  def _print_trace():
29
  traceback.print_exc()
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
 
3
  MODEL_URL = "https://huggingface.co/Bingsu/adetailer/resolve/main/deepfashion2_yolov8s-seg.pt"
 
13
  print("[INFO] DeepFashion2 model already exists.")
14
 
15
 
16
+
17
+ import traceback
18
+ from datetime import datetime
19
+ import torch, gc
20
+ from PIL import Image
21
+ import gradio as gr
22
+
23
+ from inference import generate_with_lora
24
+ from background_edit import run_background_removal_and_inpaint
25
+ from background_edit import run_clothing_inpaint
26
+
27
+
28
+
29
+
30
+
31
+
32
+
33
  # ───────────────────── Helpers ─────────────────────
34
  def _print_trace():
35
  traceback.print_exc()