Update app.py

app.py

@@ -57,11 +57,11 @@ if page == "какая-то еще":
     tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
     model = AutoModel.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
     def encode_description(description):
-        tokens = tokenizer(description, return_tensors="pt")
-        with torch.no_grad():
-            outputs = model(**tokens)
-        embeddings = outputs.last_hidden_state.mean(dim=1)
-        return embeddings
+    tokens = tokenizer(description, return_tensors="pt")
+    with torch.no_grad():
+        outputs = model(**tokens)
+    embeddings = outputs.last_hidden_state.mean(dim=1)
+    return embeddings
 
     
     embeddings = pd.read_pickle('embeddings.pkl')