Update app.py
Browse files
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 |
-
|
|
|
|
| 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
|