B4869 commited on
Commit
acf08aa
·
1 Parent(s): d495ad2

update torch version to >=2.6.0

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ model = SentenceTransformer("google/embeddinggemma-300m")
19
  def get_start():
20
  # 1. รับ JSON Payload
21
  try:
22
- query = "Hello test."
23
  query_embeddings = model.encode_query(query)
24
  query_embeddings = query_embeddings.tolist()
25
  query_embeddings_json = {
 
19
  def get_start():
20
  # 1. รับ JSON Payload
21
  try:
22
+ query = "this is a testing query."
23
  query_embeddings = model.encode_query(query)
24
  query_embeddings = query_embeddings.tolist()
25
  query_embeddings_json = {
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  Flask==3.0.0
2
  flask-cors==4.0.0
3
  sentence-transformers>=5.1.0
4
- torch==2.2.0
5
  numpy<2.0
6
  gunicorn==21.2.0
 
1
  Flask==3.0.0
2
  flask-cors==4.0.0
3
  sentence-transformers>=5.1.0
4
+ torch>=2.6.0
5
  numpy<2.0
6
  gunicorn==21.2.0