Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ from diffusers import AutoPipelineForText2Image
|
|
3 |
from generate_propmts import generate_prompt
|
4 |
from concurrent.futures import ThreadPoolExecutor, as_completed
|
5 |
from PIL import Image
|
|
|
6 |
import traceback
|
7 |
|
8 |
# Load the model once outside of the function
|
@@ -39,7 +40,6 @@ def generate_image(prompt):
|
|
39 |
traceback.print_exc()
|
40 |
return None # Return None on error to handle it gracefully in the UI
|
41 |
|
42 |
-
|
43 |
def inference(sentence_mapping, character_dict, selected_style):
|
44 |
images = []
|
45 |
print(f'sentence_mapping: {sentence_mapping}, character_dict: {character_dict}, selected_style: {selected_style}')
|
|
|
3 |
from generate_propmts import generate_prompt
|
4 |
from concurrent.futures import ThreadPoolExecutor, as_completed
|
5 |
from PIL import Image
|
6 |
+
import threading
|
7 |
import traceback
|
8 |
|
9 |
# Load the model once outside of the function
|
|
|
40 |
traceback.print_exc()
|
41 |
return None # Return None on error to handle it gracefully in the UI
|
42 |
|
|
|
43 |
def inference(sentence_mapping, character_dict, selected_style):
|
44 |
images = []
|
45 |
print(f'sentence_mapping: {sentence_mapping}, character_dict: {character_dict}, selected_style: {selected_style}')
|