Spaces:
Runtime error
Runtime error
lionelgarnier
commited on
Commit
·
cb1522f
1
Parent(s):
98b5340
simplify process_example_pipeline to return the input example prompt directly
Browse files
app.py
CHANGED
@@ -366,22 +366,8 @@ def image_to_3d(
|
|
366 |
return None, f"Error generating 3D model: {str(e)}"
|
367 |
|
368 |
|
369 |
-
|
370 |
-
|
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 |
-
|
490 |
-
fn=refine_prompt,
|
491 |
inputs=[prompt],
|
492 |
-
outputs=[
|
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(
|