Spaces:
Sleeping
Sleeping
Commit
·
685da0f
1
Parent(s):
4edd498
Update app.py
Browse files
app.py
CHANGED
@@ -42,10 +42,9 @@ d = {
|
|
42 |
|
43 |
|
44 |
if user_input and button :
|
45 |
-
predict_input = tokenizer.encode(
|
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]
|