Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,9 +9,9 @@ from concurrent.futures import ThreadPoolExecutor
|
|
9 |
# Load the model once outside of the function
|
10 |
model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
|
11 |
|
12 |
-
def generate_image(
|
13 |
try:
|
14 |
-
prompt, _ = generate_prompt(
|
15 |
image = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
|
16 |
buffered = BytesIO()
|
17 |
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
|
|
|
9 |
# Load the model once outside of the function
|
10 |
model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
|
11 |
|
12 |
+
def generate_image(sentence_mapping, character_dict, selected_style):
|
13 |
try:
|
14 |
+
prompt, _ = generate_prompt(sentence_mapping, character_dict, selected_style)
|
15 |
image = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
|
16 |
buffered = BytesIO()
|
17 |
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
|