Spaces:

fffiloni
/

HandRefiner

Running on T4

fffiloni commited on Jan 9, 2024

Commit

85f3e12

1 Parent(s): 6e701ff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,10 +3,10 @@ import subprocess
 def generate(image, prompt, seed):
     print(image, prompt, seed)
-    command = f"python handrefiner.py --input_img {image} --out_dir /content/HandRefiner/output --strength 0.55 --weights /content/HandRefiner/models/inpaint_depth_control.ckpt --prompt '{prompt}' --seed {seed}"
     try:
         result = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
-        output_path = '/content/HandRefiner/output/image_0.jpg'
         print("Output:", result.stdout)
         return output_path
     except subprocess.CalledProcessError as e:

 def generate(image, prompt, seed):
     print(image, prompt, seed)
+    command = f"python handrefiner.py --input_img {image} --out_dir /content/HandRefiner/output --strength 0.55 --weights models/inpaint_depth_control.ckpt --prompt '{prompt}' --seed {seed}"
     try:
         result = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        output_path = 'output/image_0.jpg'
         print("Output:", result.stdout)
         return output_path
     except subprocess.CalledProcessError as e: