Update app.py

app.py

@@ -21,9 +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)
-    predictions, _ = model.predict(inputs['input_ids'])
+    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)
     answer = tokenizer.decode(results[0].flatten(), skip_special_tokens=True)
     return answer