fffiloni commited on
Commit
d707478
1 Parent(s): 22e427c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ import os
4
 
5
  def generate(image, prompt, seed):
6
  print(image, prompt, seed)
7
- image_name = os.path.splitext(image)[0]
 
8
  print(f"IMAGE NAME: {image_name}")
9
  command = f"python handrefiner.py --input_img {image} --out_dir output --strength 0.55 --weights models/inpaint_depth_control.ckpt --prompt '{prompt}' --seed {seed}"
10
  try:
 
4
 
5
  def generate(image, prompt, seed):
6
  print(image, prompt, seed)
7
+ image_path = os.path.splitext(image)[0]
8
+ image_name = os.path.basename(image_path)
9
  print(f"IMAGE NAME: {image_name}")
10
  command = f"python handrefiner.py --input_img {image} --out_dir output --strength 0.55 --weights models/inpaint_depth_control.ckpt --prompt '{prompt}' --seed {seed}"
11
  try: