Update gradio_app.py

gradio_app.py

@@ -100,7 +100,7 @@ imgaedream_pipeline = None
 
 @spaces.GPU
 def gen_mvimg(
-    mvimg_model, image, seed, guidance_scale, step, text, neg_text, elevation, 
+    mvimg_model, image, seed, guidance_scale, step, text, neg_text, elevation, backgroud_color
 ):
     if seed == 0:
         seed = np.random.randint(1, 65535)
@@ -108,6 +108,8 @@ def gen_mvimg(
     if mvimg_model == "CRM":
         global crm_pipeline
         crm_pipeline.set_seed(seed)
+        background = Image.new("RGBA", image.size, (127, 127, 127))
+        image = Image.alpha_composite(background, image)
         mv_imgs = crm_pipeline(
             image, 
             scale=guidance_scale, 
@@ -117,6 +119,8 @@ def gen_mvimg(
     
     elif mvimg_model == "ImageDream":
         global imagedream_pipeline, generator
+        background = Image.new("RGBA", image.size, backgroud_color)
+        image = Image.alpha_composite(background, image)
         image = np.array(image).astype(np.float32) / 255.0
         image = image[..., :3] * image[..., 3:4] + (1 - image[..., 3:4])
         mv_imgs = imagedream_pipeline(
@@ -343,7 +347,7 @@ if __name__=="__main__":
                             outputs=[image_input]
                     ).success(
                             fn=gen_mvimg,
-                            inputs=[mvimg_model, image_input, seed, mvimg_guidance_scale, mvimg_steps, text, neg_text, elevation],
+                            inputs=[mvimg_model, image_input, seed, mvimg_guidance_scale, mvimg_steps, text, neg_text, elevation, backgroud_color],
                             outputs=[view_front, view_right, view_back, view_left]
                     ).success(
                             fn=image2mesh, 
@@ -351,7 +355,7 @@ if __name__=="__main__":
                             outputs=outputs, 
                             api_name="generate_img2obj")
         run_mv_btn.click(fn=gen_mvimg,
-                        inputs=[mvimg_model, image_input, seed, mvimg_guidance_scale, mvimg_steps, text, neg_text, elevation],
+                        inputs=[mvimg_model, image_input, seed, mvimg_guidance_scale, mvimg_steps, text, neg_text, elevation, backgroud_color],
                         outputs=[view_front, view_right, view_back, view_left]
         )
         run_3d_btn.click(fn=image2mesh,