Update app.py

app.py

@@ -64,6 +64,12 @@ def swap_faces(source_image, source_face_index, destination_image, destination_f
     result = swapper.get(destination_image, res_face, source_face, paste_back=True)
     return result
 
+async def generate_image(prompt, width, height, seed, model_name):
+    if seed == -1:
+        seed = random.randint(0, MAX_SEED)
+    image = await client.text_to_image(prompt=prompt, height=height, width=width, model=model_name)
+    return image, seed
+
 async def gen(prompt, width, height, model_name):
     seed = PREDEFINED_SEED
     image, seed = await generate_image(prompt, width, height, seed, model_name)