Error correction

app.py

@@ -20,8 +20,11 @@ def tokenize_data(text):
 
 def generate_answers(text):
     inputs = tokenize_data(text)
-    results= model.predict(inputs['input_ids'])
-    answer = tokenizer.decode(results[0], skip_special_tokens=True)
+    x = tf.dtypes.cast(inputs['input_ids'], tf.int64)
+    predictions, _ = model.predict(x)
+    results = np.argmax(predictions, axis=1)
+    print(results)
+    answer = tokenizer.decode(results[0].flatten(), skip_special_tokens=True)
     return answer
 
 examples = [["Once upon a time, "], ["In a galaxy far far away"]]