fffiloni commited on
Commit
07c4739
·
1 Parent(s): 8a5eac9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -34,12 +34,23 @@ def get_files(file_paths):
34
 
35
  def load_sfts(repo_1_id, repo_2_id):
36
 
37
- card = ModelCard.load(repo_1_id)
38
 
39
- repo_data = card.data.to_dict()
40
- instance_prompt = repo_data.get("instance_prompt")
41
- if instance_prompt is not None:
42
- print(f"Instance Prompt 1: {instance_prompt}")
 
 
 
 
 
 
 
 
 
 
 
43
 
44
 
45
  # List all ".safetensors" files in repos
 
34
 
35
  def load_sfts(repo_1_id, repo_2_id):
36
 
37
+ card_1 = ModelCard.load(repo_1_id)
38
 
39
+ repo_1_data = card_1_.data.to_dict()
40
+ instance_prompt_1 = repo_1_data.get("instance_prompt")
41
+ if instance_prompt_1 is not None:
42
+ print(f"Trigger word 1: {instance_prompt_1}")
43
+ else:
44
+ print(f"Trigger word 1: no trigger word need")
45
+
46
+ card_1 = ModelCard.load(repo_1_id)
47
+
48
+ repo_2_data = card_2_.data.to_dict()
49
+ instance_prompt_2 = repo_2_data.get("instance_prompt")
50
+ if instance_prompt_2 is not None:
51
+ print(f"Trigger word 2: {instance_prompt_2}")
52
+ else:
53
+ print(f"Trigger word 2: no trigger word need")
54
 
55
 
56
  # List all ".safetensors" files in repos