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]
|