Update app.py

app.py

@@ -101,7 +101,7 @@ def demo_retrieval():
                 ref_dev = next(model_g14.parameters()).device
                 enc = model_g14(torch.tensor(pc[:, [0, 2, 1, 3, 4, 5]].T[None], device=ref_dev)).cpu()
 
-                sim = torch.matmul(F.normalize(lvis.feats, dim=-1), F.normalize(enc, dim=-1).squeeze())
+                sim = torch.matmul(torch.nn.functional.normalize(lvis.feats, dim=-1), torch.nn.functional.normalize(enc, dim=-1).squeeze())
                 argsort = torch.argsort(sim, descending=True)
                 pred = OrderedDict((lvis.categories[i], sim[i]) for i in argsort if i < len(lvis.categories))
                 with col2: