Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def load_b_loras(content_b_lora, style_b_lora):
|
|
51 |
|
52 |
if content_b_lora is not None:
|
53 |
# Get instance_prompt a.k.a trigger word
|
54 |
-
content_model_card = ModelCard.load(
|
55 |
content_model_repo_data = content_model_card.data.to_dict()
|
56 |
content_model_instance_prompt = content_model_repo_data.get("instance_prompt")
|
57 |
else:
|
@@ -59,7 +59,7 @@ def load_b_loras(content_b_lora, style_b_lora):
|
|
59 |
|
60 |
if style_b_lora is not None:
|
61 |
# Get instance_prompt a.k.a trigger word
|
62 |
-
style_model_card = ModelCard.load(
|
63 |
style_model_repo_data = style_model_card.data.to_dict()
|
64 |
style_model_instance_prompt = style_model_repo_data.get("instance_prompt")
|
65 |
style_model_instance_prompt = f"in {style_model_instance_prompt} style"
|
|
|
51 |
|
52 |
if content_b_lora is not None:
|
53 |
# Get instance_prompt a.k.a trigger word
|
54 |
+
content_model_card = ModelCard.load(content_b_lora)
|
55 |
content_model_repo_data = content_model_card.data.to_dict()
|
56 |
content_model_instance_prompt = content_model_repo_data.get("instance_prompt")
|
57 |
else:
|
|
|
59 |
|
60 |
if style_b_lora is not None:
|
61 |
# Get instance_prompt a.k.a trigger word
|
62 |
+
style_model_card = ModelCard.load(style_b_lora)
|
63 |
style_model_repo_data = style_model_card.data.to_dict()
|
64 |
style_model_instance_prompt = style_model_repo_data.get("instance_prompt")
|
65 |
style_model_instance_prompt = f"in {style_model_instance_prompt} style"
|