Update app.py

app.py

@@ -16,13 +16,20 @@ import sentencepiece as spm
 import streamlit as st
 import matplotlib.image as mpimg
 import matplotlib.pyplot as plt
- 
+
+
 
 
  
 @st.cache(suppress_st_warning=True)
 def init():
-    torch.manual_seed(1) 
+    seed =1
+    torch.manual_seed(seed) 
+    np.random.seed(seed)
+    torch.cuda.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)
+    torch.backends.cudnn.deterministic = True
+    torch.backends.cudnn.benchmark = False
     parser = argparse.ArgumentParser()
     add_options(parser)
     args = parser.parse_args()