Update app.py

app.py

@@ -4,11 +4,11 @@
 #os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 
 import gradio as gr
-#import tensorflow as tf
+import tensorflow as tf
 import torch 
 
 import sentencepiece
-from transformers import MBartTokenizer, MBartForConditionalGeneration
+from transformers import MBartTokenizer, TFMBartForConditionalGeneration
 
 def run_model(input_text, 
               min_length,
@@ -16,13 +16,13 @@ def run_model(input_text,
               length_penalty):
 
     #MBART Transformer 
-    mbart_model =  MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50")#,from_pt=True)
+    mbart_model =  TFMBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50", from_pt=True)
     mbart_tokenizer = MBartTokenizer.from_pretrained("facebook/mbart-large-50")
 
     #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='pt')
+    input_tokenized = mbart_tokenizer.encode(input_text, return_tensors='tf')
         
     #generate input
     summary_ids = mbart_model.generate(input_tokenized,                                      
@@ -42,8 +42,8 @@ def run_model(input_text,
 
 #example 
          # source Wikipedia
-contoh = [["Dota 2 adalah sebuah permainan arena pertarungan daring multipemain, dan merupakan sekuel dari Defense of the Ancients mod pada Warcraft 3: Reign of Chaos dan Warcraft 3: The Frozen Throne. DotA 2 dikembangkan oleh Valve Corporation, terbit juli 2013 dota 2 dapat dimainkan secara gratis pada sistem operasi Microsoft Windows, OS X and Linux. Dota 2 dapat dimainkan secara eksklusif melalui distributor resmi valve, Steam.Dota 2 dimainkan oleh 2 tim yang beranggota 5 orang pemain, setiap tim memiliki markas yang berada dipojok peta, setiap markas memiliki satu bangunan bernama 'Ancient', Di mana tim harus berusaha menghancurkan 'Ancient' tim lawan agar dapat memenangkan pertandingan. Setiap pemain hanya dapat mengontrol satu karakter Hero yang berfokus pada menaikan level, mengumpulkan gold, membeli item dan melawan tim lawan untuk menang.Pengembangan Dota 2 dimulai sejak tahun 2009. Ketika pengembang mod DotA, Icefrog, dipekerjakan oleh Valve sebagai lead designer. Dota 2 dipuji oleh kritikus karena gameplay-nya, kualitas pembuatan dan kesetiaan pada gameplay pendahulu (DotA mod Warcraft 3). tetapi Dota 2 juga menuai kritik sebagai game yang susah dipelajari dan para pemain yang tidak ramah.Sampai pertengahan 2017 Dota 2 menjadi game yang memiliki aktivitas pemain paling banyak di Steam, dengan pucak 800,000 pemain online bersamaan setiap hari", 30, 300, 2], 
-          ["Elon Musk adalah pendiri beberapa perusahaan teknologi populer, termasuk SpaceX yang bergerak di teknologi antariksa; Tesla Inc. yang bergerak di bidang otomotif; OpenAI, SolarCity, Neuralink, serta beberapa perusahaan lainnya. Musk menyatakan bahwa dengan perusahaan-perusahaan ini didirikannya dengan tujuan mengubah dunia dan kehidupan manusia termasuk mengurangi pemanasan global melalui peningkatan produksi dan konsumsi energi terbarukan. Ia juga berharap dapat mengurangi 'risiko punahnya umat manusia' dengan meningkatkan teknologi untuk hidup di planet lain, khususnya di Mars. Pada bulan September 2020, kekayaan Musk ditaksir mencapai USD 102,9 miliar, menjadikannya orang terkaya ke-4 di dunia. Pada Tahun 2001, majalah Forbes menempatkannya sebagai peringkat ke-25 dalam daftar Orang Paling Berkuasa di Dunia", 30, 200, 1]]
+contoh = [["Dota 2 adalah sebuah permainan arena pertarungan daring multipemain, dan merupakan sekuel dari Defense of the Ancients mod pada Warcraft 3: Reign of Chaos dan Warcraft 3: The Frozen Throne. DotA 2 dikembangkan oleh Valve Corporation, terbit juli 2013 dota 2 dapat dimainkan secara gratis pada sistem operasi Microsoft Windows, OS X and Linux. Dota 2 dapat dimainkan secara eksklusif melalui distributor resmi valve, Steam.Dota 2 dimainkan oleh 2 tim yang beranggota 5 orang pemain, setiap tim memiliki markas yang berada dipojok peta, setiap markas memiliki satu bangunan bernama 'Ancient', Di mana tim harus berusaha menghancurkan 'Ancient' tim lawan agar dapat memenangkan pertandingan. Setiap pemain hanya dapat mengontrol satu karakter Hero yang berfokus pada menaikan level, mengumpulkan gold, membeli item dan melawan tim lawan untuk menang.Pengembangan Dota 2 dimulai sejak tahun 2009. Ketika pengembang mod DotA, Icefrog, dipekerjakan oleh Valve sebagai lead designer. Dota 2 dipuji oleh kritikus karena gameplay-nya, kualitas pembuatan dan kesetiaan pada gameplay pendahulu (DotA mod Warcraft 3). tetapi Dota 2 juga menuai kritik sebagai game yang susah dipelajari dan para pemain yang tidak ramah.Sampai pertengahan 2017 Dota 2 menjadi game yang memiliki aktivitas pemain paling banyak di Steam, dengan pucak 800,000 pemain online bersamaan setiap hari", 30, 100, 2], 
+          ["Gangguan jiwa atau penyakit jiwa adalah pola psikologis atau perilaku yang pada umumnya terkait dengan stres atau kelainan jiwa yang tidak dianggap sebagai bagian dari perkembangan normal manusia.[1] Gangguan tersebut didefinisikan sebagai kombinasi afektif, perilaku, komponen kognitif atau persepsi yang berhubungan dengan fungsi tertentu pada daerah otak atau sistem saraf yang menjalankan fungsi sosial manusia. Penemuan dan pengetahuan tentang kondisi kesehatan jiwa telah berubah sepanjang perubahan waktu dan perubahan budaya, dan saat ini masih terdapat perbedaan tentang definisi, penilaan dan klasifikasi, meskipun kriteria pedoman standar telah digunakan secara luas. Lebih dari sepertiga orang di sebagian besar negara-negara melaporkan masalah pada satu waktu pada hidup mereka yang memenuhi kriteria salah satu atau beberapa tipe umum dari kelainan jiwa.", 30, 100, 1]]
 #judul
 title = "Text Sumarization id2id"