VikramSingh178 commited on
Commit
f2cece1
1 Parent(s): c94cf04

chore: Update UI.py with base URL for API endpoints

Browse files

Former-commit-id: 21e2fce9dd108bee3a5bd5c4dfd8b33825b6b170

Files changed (2) hide show
  1. ui/ui.py +3 -2
  2. ui/vars.py +1 -0
ui/ui.py CHANGED
@@ -5,10 +5,11 @@ import json
5
  from PIL import Image
6
  from diffusers.utils import load_image
7
  from io import BytesIO
 
8
 
9
  # API endpoints
10
- sdxl_inference_endpoint = 'http://localhost:7860/api/v1/product-diffusion/sdxl_v0_lora_inference'
11
- kandinsky_inpainting_inference = 'http://localhost:7860/api/v1/product-diffusion/inpainting'
12
 
13
  def generate_sdxl_lora_image(prompt, negative_prompt, num_inference_steps, guidance_scale, num_images, mode):
14
  payload = {
 
5
  from PIL import Image
6
  from diffusers.utils import load_image
7
  from io import BytesIO
8
+ from vars import base_url
9
 
10
  # API endpoints
11
+ sdxl_inference_endpoint = f'{base_url}/api/v1/product-diffusion/sdxl_v0_lora_inference'
12
+ kandinsky_inpainting_inference = f'{base_url}/api/v1/product-diffusion/inpainting'
13
 
14
  def generate_sdxl_lora_image(prompt, negative_prompt, num_inference_steps, guidance_scale, num_images, mode):
15
  payload = {
ui/vars.py ADDED
@@ -0,0 +1 @@
 
 
1
+ base_url = 'http://localhost:7860'