Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,8 +20,7 @@ class Prodia:
|
|
| 20 |
}
|
| 21 |
|
| 22 |
def photomaker(self, params):
|
| 23 |
-
|
| 24 |
-
response = self._post(f"{self.base}/faceswap", params)
|
| 25 |
return response.json()
|
| 26 |
|
| 27 |
def get_job(self, job_id):
|
|
@@ -64,16 +63,20 @@ client = Prodia()
|
|
| 64 |
def generate_image(upload_images, prompt, negative_prompt, style_preset, steps, cfg_scale, strength, seed, progress=gr.Progress(track_tqdm=True)):
|
| 65 |
error_if_no_img(prompt)
|
| 66 |
|
| 67 |
-
|
| 68 |
"imageData": [file_to_base64(img) for img in upload_images],
|
| 69 |
"prompt": prompt,
|
| 70 |
"negative_prompt": negative_prompt,
|
| 71 |
-
"style_preset": style_preset,
|
| 72 |
"steps": steps,
|
| 73 |
"cfg_scale": cfg_scale,
|
| 74 |
"strength": strength,
|
| 75 |
"seed": seed if seed != 0 else random.randint(1, MAX_SEED)
|
| 76 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 77 |
res = prodia.wait(job)
|
| 78 |
|
| 79 |
if res['status'] == "failed":
|
|
|
|
| 20 |
}
|
| 21 |
|
| 22 |
def photomaker(self, params):
|
| 23 |
+
response = self._post(f"{self.base}/photomaker", params)
|
|
|
|
| 24 |
return response.json()
|
| 25 |
|
| 26 |
def get_job(self, job_id):
|
|
|
|
| 63 |
def generate_image(upload_images, prompt, negative_prompt, style_preset, steps, cfg_scale, strength, seed, progress=gr.Progress(track_tqdm=True)):
|
| 64 |
error_if_no_img(prompt)
|
| 65 |
|
| 66 |
+
params = {
|
| 67 |
"imageData": [file_to_base64(img) for img in upload_images],
|
| 68 |
"prompt": prompt,
|
| 69 |
"negative_prompt": negative_prompt,
|
|
|
|
| 70 |
"steps": steps,
|
| 71 |
"cfg_scale": cfg_scale,
|
| 72 |
"strength": strength,
|
| 73 |
"seed": seed if seed != 0 else random.randint(1, MAX_SEED)
|
| 74 |
+
}
|
| 75 |
+
|
| 76 |
+
if style_preset is not None and style_preset in STYLE_PRESETS:
|
| 77 |
+
params['style_preset'] = style_preset
|
| 78 |
+
|
| 79 |
+
job = client.photomaker(params)
|
| 80 |
res = prodia.wait(job)
|
| 81 |
|
| 82 |
if res['status'] == "failed":
|