lfernandopg commited on
Commit
685da0f
·
1 Parent(s): 4edd498

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -42,10 +42,9 @@ d = {
42
 
43
 
44
  if user_input and button :
45
- predict_input = tokenizer.encode([user_input],
46
  truncation=True,
47
  padding=True,
48
- max_length=512,
49
  return_tensors="tf")
50
  output = model(predict_input)[0]
51
  prediction_value = tf.argmax(output, axis=1).numpy()[0]
 
42
 
43
 
44
  if user_input and button :
45
+ predict_input = tokenizer.encode(user_input,
46
  truncation=True,
47
  padding=True,
 
48
  return_tensors="tf")
49
  output = model(predict_input)[0]
50
  prediction_value = tf.argmax(output, axis=1).numpy()[0]