andre-coy commited on
Commit
602bfa3
·
verified ·
1 Parent(s): 057c4ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -29,7 +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
 
34
  if speaker == "Surprise Me!":
35
  # load one of the provided speaker embeddings at random
 
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