mrbui1990 commited on
Commit
6182fd5
·
verified ·
1 Parent(s): 35cefdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -284,6 +284,7 @@ def infer(
284
  if expected_key not in prompt:
285
  print("❌ Invalid key.")
286
  return None
 
287
  # Load input images into PIL Images
288
  pil_images = []
289
  if images:
 
284
  if expected_key not in prompt:
285
  print("❌ Invalid key.")
286
  return None
287
+ prompt = prompt.replace(expected_key, "")
288
  # Load input images into PIL Images
289
  pil_images = []
290
  if images: