Spaces:
Running
Running
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()
|