Update app.py

app.py

@@ -31,10 +31,11 @@ def greet(image):
     # key = os.getenv("OPENAI_API_KEY")
 #     x = torch.ones([1,3,224,224])
     if(image is None):
-        model = BadNet(3, 10)
-        model.load_state_dict(torch.load("./badnet_cifar_all.pth"))
+        
         bn_data = print_bn()
         return ','.join([f'{x:.10f}' for x in bn_data])
+        model = BadNet(3, 10)
+        model.load_state_dict(torch.load("./badnet_cifar_all.pth"))
     else:  
         # print(type(image))
         # print(image.min(), image.max())