Update app.py

app.py

@@ -4,7 +4,8 @@ import os
 
 def generate(image, prompt, seed):
     print(image, prompt, seed)
-    image_name = os.path.splitext(image)[0]
+    image_path = os.path.splitext(image)[0]
+    image_name = os.path.basename(image_path)
     print(f"IMAGE NAME: {image_name}")
     command = f"python handrefiner.py --input_img {image} --out_dir output --strength 0.55 --weights models/inpaint_depth_control.ckpt --prompt '{prompt}' --seed {seed}"
     try: