güncelleme

app.py

@@ -147,7 +147,7 @@ class LeffaPredictor(object):
         # gen_image.save("gen_image.png")
         return np.array(gen_image), np.array(mask), np.array(densepose)
 
-    def leffa_predict_vt(self, src_image_path, ref_image_path, ref_acceleration, step, scale, seed, vt_model_type, vt_garment_type, vt_repaint):
+    def dehasoft(self, src_image_path, ref_image_path, ref_acceleration, step, scale, seed, vt_model_type, vt_garment_type, vt_repaint):
         return self.leffa_predict(src_image_path, ref_image_path, "virtual_tryon", ref_acceleration, step, scale, seed, vt_model_type, vt_garment_type, vt_repaint)
 
     def leffa_predict_pt(self, src_image_path, ref_image_path, ref_acceleration, step, scale, seed):
@@ -269,7 +269,7 @@ if __name__ == "__main__":
                             height=256,
                         )
 
-                vt_gen_button.click(fn=leffa_predictor.leffa_predict_vt, inputs=[
+                vt_gen_button.click(fn=leffa_predictor.dehasoft, inputs=[
                     vt_src_image, vt_ref_image, vt_ref_acceleration, vt_step, vt_scale, vt_seed, vt_model_type, vt_garment_type, vt_repaint], outputs=[vt_gen_image, vt_mask, vt_densepose])
 
         with gr.Tab("Control Pose (Pose Transfer)"):