JackismyShephard commited on
Commit
54811b2
·
1 Parent(s): f4b219c

some refactoring in application file

Browse files

fix empty input

index examples correctly

fix error not giving embedding name in example

Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -32,7 +32,11 @@ def predict(text, speaker):
32
  input_ids = inputs["input_ids"]
33
  input_ids = input_ids[..., : model.config.max_text_positions]
34
 
35
- speaker_embedding = np.load(speaker_embeddings[speaker[:3]])
 
 
 
 
36
 
37
  speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
38
 
@@ -79,7 +83,8 @@ f" checkpoint [{checkpoint_finetuned}](https://huggingface.co/{checkpoint_finetu
79
 
80
  examples = [
81
  [
82
- "I sin oprindelige før-kristne form blev alferne sandsynligvis opfattet som en personificering af det land og den natur, der omgav menneskene, dvs. den opdyrkede jord, gården og de naturressourcer, som hørte dertil. De var guddommelige eller delvis guddommelige væsener, der besad magiske kræfter, som de brugte både til fordel og ulempe for menneskene."
 
83
  ],
84
  ]
85
 
 
32
  input_ids = inputs["input_ids"]
33
  input_ids = input_ids[..., : model.config.max_text_positions]
34
 
35
+ speaker_id = speaker[:3]
36
+
37
+ speaker_embedding_path = speaker_embeddings[speaker_id]
38
+
39
+ speaker_embedding = np.load(speaker_embedding_path)
40
 
41
  speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
42
 
 
83
 
84
  examples = [
85
  [
86
+ "I sin oprindelige før-kristne form blev alferne sandsynligvis opfattet som en personificering af det land og den natur, der omgav menneskene, dvs. den opdyrkede jord, gården og de naturressourcer, som hørte dertil. De var guddommelige eller delvis guddommelige væsener, der besad magiske kræfter, som de brugte både til fordel og ulempe for menneskene.",
87
+ "F23 (Female, 23, Vestjylland)",
88
  ],
89
  ]
90