andre-coy commited on
Commit
f647915
โ€ข
1 Parent(s): 602bfa3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -29,9 +29,8 @@ def predict(text, speaker):
29
 
30
  # limit input length
31
  input_ids = inputs["input_ids"]
32
- #input_ids = input_ids[..., :model.config.max_text_positions]
33
- input_ids = input_ids[..., :5000]
34
-
35
  if speaker == "Surprise Me!":
36
  # load one of the provided speaker embeddings at random
37
  idx = np.random.randint(len(speaker_embeddings))
 
29
 
30
  # limit input length
31
  input_ids = inputs["input_ids"]
32
+ input_ids = input_ids[..., :model.config.max_text_positions]
33
+
 
34
  if speaker == "Surprise Me!":
35
  # load one of the provided speaker embeddings at random
36
  idx = np.random.randint(len(speaker_embeddings))