Update app.py

app.py

@@ -394,7 +394,7 @@ def file_upload(file, net):
     weights = torch.load(file.name).to(device)
         
     #pad to 10000 Principal components to keep everything consistent
-    pcs = net.shape[1]
+    pcs = weights.shape[1]
     padding =  torch.zeros((1,10000-pcs)).to(device)
     weights = torch.cat((weights, padding), 1)
     net = "model_"+str(uuid.uuid4())[:4]+".pt"