Update app.py

app.py

@@ -13,7 +13,7 @@ def tokenize_data(text):
     input_ =  str(text) + ' </s>'
     max_len = 80
     # tokenize inputs
-    tokenized_inputs = tokenizer(input_, padding='max_length', truncation=True, max_length=max_len, return_attention_mask=True)
+    tokenized_inputs = tokenizer(input_, padding='max_length', truncation=True, max_length=max_len, return_attention_mask=True, return_tensors='pt')
 
     inputs={"input_ids": tokenized_inputs['input_ids'],
         "attention_mask": tokenized_inputs['attention_mask']}