random seed

app.py

@@ -8,7 +8,7 @@ device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
 # model=torch.load("Gpt_neo_Epoch_10_Loss_031_data_5000.pth",map_location=torch.device('cpu'))
-
+torch.manual_seed(0)
 model=torch.load("Gpt_neo_Epoch_10_Loss_031_data_5000.pth",map_location=device)
 
 def predict_query(input_sentence,max_len=40,temp=0.7):
@@ -68,6 +68,7 @@ if Enter:
     st.header("Output-")
     print("Generating predictions......\n\n")
     # out=generate(sentence,max_len,top_k,do_sample)
+    torch.manual_seed(0)
     out=predict_query(sentence,max_len,temp)
     for i,out in enumerate(out):
         st.markdown(f"Query {i} :{out}")