feylur commited on
Commit
6d95b71
·
verified ·
1 Parent(s): 1f9c7f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -94,7 +94,7 @@ def generate_tryon(person_img, cloth_img, progress=gr.Progress()):
94
  # Clear memory
95
  gc.collect()
96
 
97
- progress(0.50, desc="⚠️ **Note:** Running on GPU T4 - processing takes 2-3 minutes per image.")
98
 
99
  # Run inference
100
  result = pipeline(
@@ -125,7 +125,7 @@ with gr.Blocks(
125
  # 🎨 Try-Space Virtual Try-On
126
  ### Upload a person image and garment to see the magic! ✨
127
 
128
- ⚠️ **Note:** Running on CPU - processing takes 5-10 minutes per image.
129
  """)
130
 
131
  with gr.Row():
@@ -170,8 +170,7 @@ with gr.Blocks(
170
  - ✅ Avoid extreme angles or poses
171
 
172
  ### ⏱️ Processing Time:
173
- - **CPU Basic:** ~5-10 minutes per generation
174
- - **Upgrade to GPU:** Reduces to ~2-3 minutes
175
  """)
176
 
177
  # Event handler
@@ -183,9 +182,7 @@ with gr.Blocks(
183
 
184
  # Launch app
185
  if __name__ == "__main__":
186
- print("🚀 Starting Try-Space Virtual Try-On...")
187
- print("⏳ First run will download models (~2-3 GB)...")
188
-
189
  # Pre-load models at startup
190
  try:
191
  load_models()
 
94
  # Clear memory
95
  gc.collect()
96
 
97
+ progress(0.50, desc="Running on GPU T4 - processing takes 2-3 minutes per image.")
98
 
99
  # Run inference
100
  result = pipeline(
 
125
  # 🎨 Try-Space Virtual Try-On
126
  ### Upload a person image and garment to see the magic! ✨
127
 
128
+ ⚠️ **Note:** Running on GPU T4 - processing takes 2-3 minutes per image.
129
  """)
130
 
131
  with gr.Row():
 
170
  - ✅ Avoid extreme angles or poses
171
 
172
  ### ⏱️ Processing Time:
173
+ - **GPU T4:** ~2-3 minutes per generation
 
174
  """)
175
 
176
  # Event handler
 
182
 
183
  # Launch app
184
  if __name__ == "__main__":
185
+ print("🚀 Starting Try-Space Virtual Try-On...")
 
 
186
  # Pre-load models at startup
187
  try:
188
  load_models()