Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,9 @@ kii=" mohawk femboy racecar driver "
|
|
12 |
|
13 |
|
14 |
|
|
|
|
|
|
|
15 |
def log_message():
|
16 |
logs = ""
|
17 |
for i in range(5): # Simulate some logging over time
|
@@ -70,7 +73,7 @@ def gen_fn(model_str, prompt, negative_prompt):
|
|
70 |
try:
|
71 |
# Attempt to generate the image
|
72 |
image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
|
73 |
-
|
74 |
# Check if the image_response is a tuple, handle accordingly
|
75 |
if isinstance(image_response, tuple):
|
76 |
# If the response is a tuple, assume the first item is the image
|
@@ -85,7 +88,7 @@ def gen_fn(model_str, prompt, negative_prompt):
|
|
85 |
print(f"Unexpected response type: {type(image_response)}")
|
86 |
return None
|
87 |
except Exception as e:
|
88 |
-
print(f"
|
89 |
return None
|
90 |
|
91 |
|
|
|
12 |
|
13 |
|
14 |
|
15 |
+
|
16 |
+
|
17 |
+
|
18 |
def log_message():
|
19 |
logs = ""
|
20 |
for i in range(5): # Simulate some logging over time
|
|
|
73 |
try:
|
74 |
# Attempt to generate the image
|
75 |
image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
|
76 |
+
print(f"{image_response}")
|
77 |
# Check if the image_response is a tuple, handle accordingly
|
78 |
if isinstance(image_response, tuple):
|
79 |
# If the response is a tuple, assume the first item is the image
|
|
|
88 |
print(f"Unexpected response type: {type(image_response)}")
|
89 |
return None
|
90 |
except Exception as e:
|
91 |
+
print(f"Error occurred: {e}")
|
92 |
return None
|
93 |
|
94 |
|