Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ def infer(
|
|
| 50 |
steps: int,
|
| 51 |
guidance_scale: float,
|
| 52 |
) -> dict:
|
| 53 |
-
"""Generates an image from a scribble using FLUX.2 Klein 9B."""
|
| 54 |
gc.collect()
|
| 55 |
torch.cuda.empty_cache()
|
| 56 |
|
|
@@ -58,7 +58,7 @@ def infer(
|
|
| 58 |
raise gr.Error("Please enter a prompt.")
|
| 59 |
|
| 60 |
if not image_b64:
|
| 61 |
-
raise gr.Error("Please draw a sketch first.")
|
| 62 |
|
| 63 |
try:
|
| 64 |
header, data = image_b64.split(",", 1)
|
|
|
|
| 50 |
steps: int,
|
| 51 |
guidance_scale: float,
|
| 52 |
) -> dict:
|
| 53 |
+
"""Generates an image from a scribble (with assets composited) using FLUX.2 Klein 9B."""
|
| 54 |
gc.collect()
|
| 55 |
torch.cuda.empty_cache()
|
| 56 |
|
|
|
|
| 58 |
raise gr.Error("Please enter a prompt.")
|
| 59 |
|
| 60 |
if not image_b64:
|
| 61 |
+
raise gr.Error("Please draw a sketch or add an asset first.")
|
| 62 |
|
| 63 |
try:
|
| 64 |
header, data = image_b64.split(",", 1)
|