Update app.py

app.py

@@ -21,8 +21,9 @@ def tokenize_data(text):
 
 def generate_answers(text):
     inputs = tokenize_data(text)
-    x = tf.dtypes.cast(inputs['input_ids'], tf.int64)
-    predictions, _ = model.predict(x)
+    #x = tf.dtypes.cast(inputs['input_ids'], tf.int64)
+    #predictions, _ = model.predict(x)
+    predictions, _ = model.predict(inputs['input_ids'])
     results = np.argmax(predictions, axis=1)
     print(results)
     answer = tokenizer.decode(results[0].flatten(), skip_special_tokens=True)