Update app.py
Browse files
app.py
CHANGED
@@ -22,6 +22,10 @@ def extract_metadata(image):
|
|
22 |
if 'sui_image_params' in parameters_data:
|
23 |
sui_image_params = parameters_data['sui_image_params']
|
24 |
metadata.update(sui_image_params)
|
|
|
|
|
|
|
|
|
25 |
else:
|
26 |
lines = image.info['parameters'].split('\n')
|
27 |
prompt = lines[0].strip()
|
|
|
22 |
if 'sui_image_params' in parameters_data:
|
23 |
sui_image_params = parameters_data['sui_image_params']
|
24 |
metadata.update(sui_image_params)
|
25 |
+
else:
|
26 |
+
metadata = parameters_data
|
27 |
+
elif 'num_inference_steps' in parameters_data:
|
28 |
+
|
29 |
else:
|
30 |
lines = image.info['parameters'].split('\n')
|
31 |
prompt = lines[0].strip()
|