Update app.py

app.py

@@ -52,27 +52,11 @@ downscaler = FnNode(
    },
 )
 
-flux_enhancer = InferenceNode(
-   model="black-forest-labs/FLUX.2-klein-4B:replicate",
-   inputs={
-       "image": downscaler.image,
-       "prompt": gr.Textbox(
-           label="prompt",
-           value=("Transform this into a clean 3D asset render"),
-           lines=3,
-       ),
-   },
-   outputs={
-       "image": gr.Image(label="3D-Ready Enhanced Image"),
-   },
-)
-
-
 trellis_3d = GradioNode(
    "microsoft/TRELLIS.2",
    api_name="/image_to_3d",
    inputs={
-       "image": flux_enhancer.image,
+       "image": downscaler.image,
        "ss_guidance_strength": 7.5,   
        "ss_sampling_steps": 12,     
    },
@@ -83,7 +67,7 @@ trellis_3d = GradioNode(
 
 graph = Graph(
    name="Image to 3D Asset Pipeline",
-   nodes=[background_remover, downscaler, flux_enhancer, trellis_3d],
+   nodes=[background_remover, downscaler, trellis_3d],
 )