add api

app.py

@@ -15,6 +15,7 @@ from transformers import CLIPTextModel, CLIPTokenizer
 from briarmbg import BriaRMBG
 from enum import Enum
 from torch.hub import download_url_to_file
+from loadimg import load_img
 
 
 # 'stablediffusionapi/realistic-vision-v51'
@@ -332,6 +333,10 @@ def process_relight(input_fg, input_bg, prompt, image_width, image_height, num_s
     results = [(x * 255.0).clip(0, 255).astype(np.uint8) for x in results]
     return results + extra_images
 
+def fn(fg, bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source):
+    input_fg = load_img(fg, output_type="numpy")
+    input_bg = load_img(bg, output_type="numpy")
+    return process_relight(input_fg, input_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source)
 
 @spaces.GPU
 @torch.inference_mode()
@@ -463,6 +468,8 @@ with block:
         return gal[evt.index]['name']
 
     bg_gallery.select(bg_gallery_selected, inputs=bg_gallery, outputs=input_bg)
+    
+    gr.Interface(fn, inputs=ips, outputs=[result_gallery], api_name="fn")
 
 
 block.launch(server_name='0.0.0.0')

requirements.txt

@@ -8,3 +8,4 @@ torch
 peft
 gradio
 protobuf
+loadimg