add text prompt

app.py

@@ -49,7 +49,7 @@ sys.path.append(os.path.join(MV_ADAPTER_CODE_DIR, "scripts"))
 HEADER = """
 # <img src="https://compass.uol/content/dam/aircompanycompass/header/logo-desktop.png" alt="Compass.UOL">
 
-# Image to 3D | Nestlé | Proof of Concept
+# Compass.UOL | Nestlé| Image to 3D | Proof of Concept
 
 ## State-of-the-art 3D Generation Using Large-Scale Rectified Flow Transformers
 
@@ -286,7 +286,7 @@ def image_to_3d(
 
 @spaces.GPU(duration=120)
 @torch.no_grad()
-def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
+def run_texture(image: Image, mesh_path: str, seed: int, text_prompt: str, req: gr.Request):
     height, width = 768, 768
     # Prepare cameras
     cameras = get_orthogonal_camera(
@@ -332,7 +332,7 @@ def run_texture(image: Image, mesh_path: str, seed: int, req: gr.Request):
         pipe_kwargs["generator"] = torch.Generator(device=DEVICE).manual_seed(seed)
 
     images = mv_adapter_pipe(
-        "high quality",
+        text_prompt,
         height=height,
         width=width,
         num_inference_steps=15,
@@ -386,6 +386,7 @@ with gr.Blocks(title="Nestlé | Proof of Concept") as demo:
                 )
 
             with gr.Accordion("Generation Settings", open=True):
+                text_prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt", value="high quality")
                 seed = gr.Slider(
                     label="Seed",
                     minimum=0,
@@ -455,7 +456,7 @@ with gr.Blocks(title="Nestlé | Proof of Concept") as demo:
 
     gen_texture_button.click(
         run_texture,
-        inputs=[image_prompts, model_output, seed],
+        inputs=[image_prompts, model_output, seed, text_prompt],
         outputs=[textured_model_output]
     )