This commit is contained in:
filnow 2024-04-15 12:20:41 +02:00
commit 5ed603bafd
2 changed files with 40 additions and 10 deletions

View File

@ -1,10 +1,5 @@
user: Dzień dobry user Dzień dobry hello()
system: Czesc, jestem agentem dialogowym przyjmujacym zamowienia w restauracji. system Czesc, jestem agentem dialogowym przyjmujacym zamowienia w restauracji. Moge doradzic ci w wyborze odpowiedniej pozycji z menu. W czym moge ci pomoc? welcomemsg()
Moge doradzic ci w wyborze odpowiedniej pozycji z menu. user Chciałbym zjeść 3-daniowy obiad z kuchni włoskiej. inform()
W czym moge ci pomoc? system Proponuję "Sałatka caprese", "Lasagne", "Tiramisu" select()
user: Chciałbym zjeść 3-daniowy obiad z kuchni włoskiej. user Mógłbyś mi napisać z jakich składników składa się sałatka? reqmore()
system: Proponuję
"Sałatka caprese",
"Lasagne",
"Tiramisu"
user: Mógłbyś mi napisać z jakich składników składa się sałatka?
Can't render this file because it contains an unexpected character in line 7 and column 18.

35
embeddings.py Normal file
View File

@ -0,0 +1,35 @@
import numpy as np
import torch
from transformers import AutoModel, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("Geotrend/distilbert-base-pl-cased")
model = AutoModel.from_pretrained("Geotrend/distilbert-base-pl-cased")
text = """
"nazwa": "Tatar wołowy","""
# "skladniki": [
# "wołowina",
# "cebula",
# "ogórki kiszone",
# "musztarda",
# "jajko",
# "pieprz",
# "sól"
# ],
# "alergeny": [
# "jajko",
# "gorczyca"
# ]
# """
encoded_input = tokenizer(text, return_tensors='pt', padding=True)
output = model(**encoded_input)
prompt = "tatar"
encoded_prompt = tokenizer(prompt, return_tensors='pt', padding=True)
output_prompt = model(**encoded_prompt)
text_embedding = output.last_hidden_state[:, 0, :]
prompt_embedding = output_prompt.last_hidden_state[:, 0, :]
cosine = torch.nn.functional.cosine_similarity(
text_embedding, prompt_embedding, dim=1)
print(cosine.item())