Update app.py

app.py

@@ -9,19 +9,34 @@ model.train()
 
 import os 
 
+def print_bn():
+    bn_data = []
+    for m in model.modules():
+        if(type(m) is nn.BatchNorm2d):
+            # print(m.momentum)
+            bn_data.extend(m.running_mean.data.numpy().tolist())
+            bn_data.extend(m.running_var.data.numpy().tolist())
+            bn_data.append(float(m.momentum.data))
+    return bn_data
+
 def greet(image):
     # url = f'https://huggingface.co/spaces?p=1&sort=modified&search=GPT'
     # html = request_url(url)
     # key = os.getenv("OPENAI_API_KEY")
 #     x = torch.ones([1,3,224,224])
-    print(type(image))
-    image = torch.tensor(image).float()
-    print(image.min(), image.max())
-    image = image/255.0
-    image = torch.permute(image, [0,2,3,1])
-    out = model(image)
+    if(image is None):
+        print_bn()
+    else:
+        print(type(image))
+        image = torch.tensor(image).float()
+        print(image.min(), image.max())
+        image = image/255.0
+        image = image.unsqueeze(0)
+        image = torch.permute(image, [0,2,3,1])
+        out = model(image)
+
     # model.train()
-    return f"Hello {model.bn1.running_mean.data} !!"
+    return f"Hello YOU !!"