update app

app.py

@@ -105,8 +105,7 @@ class LeffaPredictor(object):
         # gen_image.save("gen_image.png")
         return np.array(gen_image)
 
-    def leffa_predict_vt(self, src_image_path, ref_image_path, step, scale, seed, vt_model_type="viton_hd"):
-        self.change_vt_model(vt_model_type)
+    def leffa_predict_vt(self, src_image_path, ref_image_path, step, scale, seed):
         return self.leffa_predict(src_image_path, ref_image_path, "virtual_tryon", step, scale, seed)
 
     def leffa_predict_pt(self, src_image_path, ref_image_path, step, scale, seed):
@@ -192,14 +191,8 @@ if __name__ == "__main__":
                         vt_seed = gr.Number(
                             label="Random Seed", minimum=-1, maximum=2147483647, step=1, value=42)
 
-                        vt_model_type = gr.Radio(
-                            choices=["viton_hd", "dress_code"],
-                            value="viton_hd",
-                            label="Model Type",
-                        )
-
                 vt_gen_button.click(fn=leffa_predictor.leffa_predict_vt, inputs=[
-                    vt_src_image, vt_ref_image, vt_step, vt_scale, vt_seed, vt_model_type], outputs=[vt_gen_image])
+                    vt_src_image, vt_ref_image, vt_step, vt_scale, vt_seed], outputs=[vt_gen_image])
 
         with gr.Tab("Control Pose (Pose Transfer)"):
             with gr.Row():