Update app.py

app.py

@@ -7,6 +7,14 @@ from torchvision import transforms
 import random
 import numpy as np
 
+def set_seed(seed):
+    random.seed(seed)
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    if torch.cuda.is_available():
+        torch.cuda.manual_seed(seed)
+        torch.cuda.manual_seed_all(seed)
+        
 # Load model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -81,6 +89,8 @@ def IQA_preprocess():
     ])
     return transform
 
+set_seed(3407)
+
 # Inference function
 def predict_percept(image):
     # If the image is passed as a PIL Image