Update script.py

adding test to make sure network access is blocked

script.py

@@ -23,7 +23,15 @@ from preprocess import preprocess
 DATASET_PATH = "/tmp/data"
 dataset_remote = load_dataset(DATASET_PATH,split = "test",streaming = True)
 
-
+# this should fail since there is not network access during model run
+try:
+    import requests
+    r = requests.get("https://stresearch.github.io/SAFE/logo.jpg")
+    print(r.text)
+except Exception as e:
+    print("download test faild as expected")
+    print(e)
+    
 # load your model
 device = "cuda:0"
 model = Model().to(device)