Update main.py

main.py

@@ -18,7 +18,7 @@ client = prodia.Client(api_key=prodia_key)
 def process_input_text2img(prompt, negative_prompt, steps, cfg_scale, number, seed, model, sampler, aspect_ratio, upscale, save=False):
     images = []
     for image in range(number):
-        result = client.txt2img(prompt=prompt, negative_prompt=negative_prompt, model=model, sampler=sampler,
+        result = client.sd_generate(prompt=prompt, negative_prompt=negative_prompt, model=model, sampler=sampler,
                                 steps=steps, cfg_scale=cfg_scale, seed=seed, aspect_ratio=aspect_ratio, upscale=upscale)
         images.append(result.url)
         if save:
@@ -33,7 +33,7 @@ def process_input_text2img(prompt, negative_prompt, steps, cfg_scale, number, se
 def process_input_img2img(init, prompt, negative_prompt, steps, cfg_scale, number, seed, model, sampler, ds, upscale, save):
     images = []
     for image in range(number):
-        result = client.img2img(imageUrl=init, prompt=prompt, negative_prompt=negative_prompt, model=model, sampler=sampler,
+        result = client.sd_transform(imageUrl=init, prompt=prompt, negative_prompt=negative_prompt, model=model, sampler=sampler,
                                 steps=steps, cfg_scale=cfg_scale, seed=seed, denoising_strength=ds, upscale=upscale)
         images.append(result.url)
         if save: