fffiloni commited on
Commit
fd52df0
·
verified ·
1 Parent(s): 7af6efa

Update app.py

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