Update app.py

app.py

@@ -4,8 +4,8 @@
 #os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 import gradio as gr
-#import tensorflow as tf
-#import torch 
+import tensorflow as tf
+import torch 
 
 import sentencepiece
 from transformers import MBartTokenizer, MBartForConditionalGeneration
@@ -22,7 +22,7 @@ def run_model(input_text,
     #encode input to vector
     input_text = str(input_text)
     input_text = ' '.join(input_text.split()) # hapus white space 
-    input_tokenized = mbart_tokenizer.encode(input_text, return_tensors='np')
+    input_tokenized = mbart_tokenizer.encode(input_text, return_tensors='pt')
         
     #generate input
     summary_ids = mbart_model.generate(input_tokenized,