Update README.md
Browse files
README.md
CHANGED
@@ -32,7 +32,7 @@ food_data = "Hawaiian Pizza"
|
|
32 |
embedding = pipeline(
|
33 |
'feature-extraction', model='alexdseo/RecipeBERT', framework='pt'
|
34 |
)
|
35 |
-
#
|
36 |
food_rep = embedding(food_data, return_tensors='pt')[0].numpy().mean(axis=0)
|
37 |
|
38 |
```
|
|
|
32 |
embedding = pipeline(
|
33 |
'feature-extraction', model='alexdseo/RecipeBERT', framework='pt'
|
34 |
)
|
35 |
+
# Mean pooling
|
36 |
food_rep = embedding(food_data, return_tensors='pt')[0].numpy().mean(axis=0)
|
37 |
|
38 |
```
|