VOIDER commited on
Commit
a16cc05
·
verified ·
1 Parent(s): b1862d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -48,11 +48,12 @@ def extract_metadata(file):
48
  parser = ParserManager()
49
  info = parser.parse(file.name)
50
  prompt = info.prompts[0].value if info.prompts else ''
 
 
51
  if hasattr(info, 'models') and info.models:
52
- first = info.models[0]
 
53
  model_name = first.name if hasattr(first, 'name') else str(first)
54
- else:
55
- model_name = ''
56
  return prompt, model_name
57
 
58
  # Image preprocessing transform
 
48
  parser = ParserManager()
49
  info = parser.parse(file.name)
50
  prompt = info.prompts[0].value if info.prompts else ''
51
+ # info.models may be a set or list of model identifiers
52
+ model_name = ''
53
  if hasattr(info, 'models') and info.models:
54
+ # pick one representative model
55
+ first = next(iter(info.models))
56
  model_name = first.name if hasattr(first, 'name') else str(first)
 
 
57
  return prompt, model_name
58
 
59
  # Image preprocessing transform