Spaces:
Sleeping
Sleeping
Update classification.py
Browse files- classification.py +1 -1
classification.py
CHANGED
@@ -138,7 +138,7 @@ def initialize_models():
|
|
138 |
model_ST = SentenceTransformer("all-mpnet-base-v2",device = "cuda")
|
139 |
return model_ST
|
140 |
|
141 |
-
|
142 |
def generate_embeddings(df, model, Column):
|
143 |
embeddings_list = []
|
144 |
for index, row in df.iterrows():
|
|
|
138 |
model_ST = SentenceTransformer("all-mpnet-base-v2",device = "cuda")
|
139 |
return model_ST
|
140 |
|
141 |
+
@spaces.GPU
|
142 |
def generate_embeddings(df, model, Column):
|
143 |
embeddings_list = []
|
144 |
for index, row in df.iterrows():
|