andre-coy commited on
Commit
311b2f7
·
verified ·
1 Parent(s): 8405e64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def predict(text, speaker):
28
  inputs = processor(text=text, return_tensors="pt")
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!":
 
28
  inputs = processor(text=text, return_tensors="pt")
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!":