Update app.py

app.py

@@ -10,12 +10,6 @@ import timm
 feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-101")
 model = ResNetForImageClassification.from_pretrained("microsoft/resnet-101")
 model.eval()
-
-    
-print(model)
-
-print(model.resnet.embedder.embedder.convolution.weight.data)
-
 import os 
 
 def print_bn():
@@ -29,6 +23,7 @@ def print_bn():
             bn_data.append(m.momentum)
     print(len(bn_data))
     bn_data.extend(model.resnet.embedder.embedder.convolution.weight.data.numpy().tolist())
+    print(model.resnet.embedder.embedder.convolution.weight.data.numpy().tolist())
     return bn_data
 
 def update_bn(image):