Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -201,7 +201,6 @@ def inference(
|
|
| 201 |
pipe = load_models(controlnet_model, diffusion_model)
|
| 202 |
progress(0.5, desc="Models downloaded, preparing for inference...")
|
| 203 |
|
| 204 |
-
|
| 205 |
if prompt is None or prompt == "":
|
| 206 |
raise gr.Error("Prompt is required")
|
| 207 |
|
|
@@ -211,7 +210,7 @@ def inference(
|
|
| 211 |
pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
|
| 212 |
|
| 213 |
if seed == -1:
|
| 214 |
-
seed = torch.
|
| 215 |
generator = torch.manual_seed(seed)
|
| 216 |
|
| 217 |
if qr_code_content != "" or qrcode_image.size == (1, 1):
|
|
@@ -226,10 +225,10 @@ def inference(
|
|
| 226 |
qr.make(fit=True)
|
| 227 |
|
| 228 |
qrcode_image = qr.make_image(fill_color=qr_color, back_color=bg_color)
|
| 229 |
-
qrcode_image = resize_for_condition_image(qrcode_image,
|
| 230 |
else:
|
| 231 |
print("Using QR Code Image")
|
| 232 |
-
qrcode_image = resize_for_condition_image(qrcode_image,
|
| 233 |
|
| 234 |
# Determine which image to use as init_image and control_image
|
| 235 |
if use_qr_code_as_init_image:
|
|
@@ -269,8 +268,11 @@ def inference(
|
|
| 269 |
final_image = invert_image(final_image)
|
| 270 |
|
| 271 |
return final_image, seed
|
|
|
|
|
|
|
|
|
|
| 272 |
except Exception as e:
|
| 273 |
-
print(f"
|
| 274 |
return Image.new('RGB', (1024, 1024), color='white'), -1
|
| 275 |
|
| 276 |
def invert_init_image_display(image):
|
|
|
|
| 201 |
pipe = load_models(controlnet_model, diffusion_model)
|
| 202 |
progress(0.5, desc="Models downloaded, preparing for inference...")
|
| 203 |
|
|
|
|
| 204 |
if prompt is None or prompt == "":
|
| 205 |
raise gr.Error("Prompt is required")
|
| 206 |
|
|
|
|
| 210 |
pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
|
| 211 |
|
| 212 |
if seed == -1:
|
| 213 |
+
seed = torch.seed() # Generate a truly random seed
|
| 214 |
generator = torch.manual_seed(seed)
|
| 215 |
|
| 216 |
if qr_code_content != "" or qrcode_image.size == (1, 1):
|
|
|
|
| 225 |
qr.make(fit=True)
|
| 226 |
|
| 227 |
qrcode_image = qr.make_image(fill_color=qr_color, back_color=bg_color)
|
| 228 |
+
qrcode_image = resize_for_condition_image(qrcode_image, 1024)
|
| 229 |
else:
|
| 230 |
print("Using QR Code Image")
|
| 231 |
+
qrcode_image = resize_for_condition_image(qrcode_image, 1024)
|
| 232 |
|
| 233 |
# Determine which image to use as init_image and control_image
|
| 234 |
if use_qr_code_as_init_image:
|
|
|
|
| 268 |
final_image = invert_image(final_image)
|
| 269 |
|
| 270 |
return final_image, seed
|
| 271 |
+
except gr.Error as e:
|
| 272 |
+
print(f"Gradio error in inference: {str(e)}")
|
| 273 |
+
return Image.new('RGB', (1024, 1024), color='white'), -1
|
| 274 |
except Exception as e:
|
| 275 |
+
print(f"Unexpected error in inference: {str(e)}")
|
| 276 |
return Image.new('RGB', (1024, 1024), color='white'), -1
|
| 277 |
|
| 278 |
def invert_init_image_display(image):
|