theoracle commited on
Commit
71f1758
Β·
1 Parent(s): ab47115

fix crash app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -8
app.py CHANGED
@@ -1,12 +1,4 @@
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, run_clothing_inpaint
10
 
11
  # Ensure DeepFashion2 model is downloaded early
12
  MODEL_URL = "https://huggingface.co/Bingsu/adetailer/resolve/main/deepfashion2_yolov8s-seg.pt"
@@ -17,6 +9,18 @@ if not os.path.exists(MODEL_PATH):
17
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
18
  print("[INFO] Model downloaded.")
19
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  # ─────────────── Helpers ───────────────
21
  def _print_trace():
22
  traceback.print_exc()
 
1
  import os
 
 
 
 
 
 
 
 
2
 
3
  # Ensure DeepFashion2 model is downloaded early
4
  MODEL_URL = "https://huggingface.co/Bingsu/adetailer/resolve/main/deepfashion2_yolov8s-seg.pt"
 
9
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
10
  print("[INFO] Model downloaded.")
11
 
12
+
13
+
14
+ import traceback
15
+ from datetime import datetime
16
+ import torch, gc
17
+ from PIL import Image
18
+ import gradio as gr
19
+
20
+ from inference import generate_with_lora
21
+ from background_edit import run_background_removal_and_inpaint, run_clothing_inpaint
22
+
23
+
24
  # ─────────────── Helpers ───────────────
25
  def _print_trace():
26
  traceback.print_exc()