prithivMLmods commited on
Commit
3c36366
·
verified ·
1 Parent(s): 92e41c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)