Update app.py

app.py

@@ -35,9 +35,8 @@ from hair_swap import HairFast, get_parser
 hair_fast = HairFast(get_parser().parse_args([]))
 
 @spaces.GPU
-def swap_hair(source, target_1, target_2):
-    result = hair_fast(face_img, shape_img, color_img)
-    final_image = hair_fast.swap(face_path, shape_path, color_path)
+def swap_hair(source, target_1, target_2, progress=gr.Progress(track_tqdm=True)):
+    final_image = hair_fast.swap(source, target_1, target_2)
     return T.functional.to_pil_image(final_image)
     
 with gr.Blocks() as demo: