Linaqruf commited on
Commit
c30b9c5
1 Parent(s): 4ca2787

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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()