add streamlit

app.py

@@ -1,6 +1,7 @@
 import os
 import torch
 import argparse
+import streamlit as st
 import sentencepiece as spm
 from utils import utils_cls
 from model import BanglaTransformer
@@ -83,7 +84,7 @@ class Bn2EnTranslation:
             "bn_tokenizer" : bn_tokenizer, 
             "bn_vocab" : bn_vocab,
             "en_vocab" :  en_vocab, 
-            "model" : model
+            "model": model
         }
         return models 
     def translate(self, text, models):
@@ -105,14 +106,14 @@ class Bn2EnTranslation:
 
 
 
-if __name__ == "__main__":
-    print(torch.cuda.get_device_name(0))
-    text = "এই উপজেলায় ১টি সরকারি কলেজ রয়েছে"
-
-    obj = Bn2EnTranslation()
-    models = obj.get_bntoen_model()
+# if __name__ == "__main__":
+#     print(torch.cuda.get_device_name(0))
+text = "এই উপজেলায় ১টি সরকারি কলেজ রয়েছে"
+obj = Bn2EnTranslation()
+models = obj.get_bntoen_model()
+text = st.text_area("Enter some text:এই উপজেলায় ১টি সরকারি কলেজ রয়েছে")
+if text:
     pre = obj.translate(text, models)
-    print("="*20)
-    print(f"input : {text}")
-    print(f"prediction: {pre}")
+    print(f"Input      : {text}")
+    print(f"Prediction : {pre}")