lionelgarnier commited on
Commit
cb1522f
·
1 Parent(s): 98b5340

simplify process_example_pipeline to return the input example prompt directly

Browse files
Files changed (1) hide show
  1. app.py +4 -23
app.py CHANGED
@@ -366,22 +366,8 @@ def image_to_3d(
366
  return None, f"Error generating 3D model: {str(e)}"
367
 
368
 
369
- # Create a combined function that handles the whole pipeline from example to image
370
- # This version gets the parameters from the UI components
371
- @spaces.GPU()
372
- def process_example_pipeline(example_prompt, system_prompt=DEFAULT_SYSTEM_PROMPT, progress=gr.Progress()):
373
- # Step 1: Update status
374
- progress(0, desc="Starting example processing")
375
-
376
- # Step 2: Refine the prompt
377
- progress(0.1, desc="Refining prompt with Mistral")
378
- refined, status = refine_prompt(example_prompt, system_prompt, progress)
379
-
380
- if not refined:
381
- return "", "Failed to refine prompt: " + status
382
-
383
- # Return only the refined prompt and status - don't generate image
384
- return refined, "Prompt refined successfully!"
385
 
386
  def create_interface():
387
  # Preload models if needed
@@ -486,15 +472,10 @@ def create_interface():
486
 
487
  gr.Examples(
488
  examples=examples,
489
- # fn=process_example_pipeline,
490
- fn=refine_prompt,
491
  inputs=[prompt],
492
- outputs=[refined_prompt, message_box],
493
  cache_examples=True,
494
- ).then(
495
- fn=generate_image,
496
- inputs=[refined_prompt, flux_seed, flux_randomize_seed, width, height, num_inference_steps],
497
- outputs=[generated_image, message_box]
498
  )
499
 
500
  gr.on(
 
366
  return None, f"Error generating 3D model: {str(e)}"
367
 
368
 
369
+ def process_example_pipeline(example_prompt):
370
+ return example_prompt
 
 
 
 
 
 
 
 
 
 
 
 
 
 
371
 
372
  def create_interface():
373
  # Preload models if needed
 
472
 
473
  gr.Examples(
474
  examples=examples,
475
+ fn=process_example_pipeline,
 
476
  inputs=[prompt],
477
+ outputs=[prompt],
478
  cache_examples=True,
 
 
 
 
479
  )
480
 
481
  gr.on(