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():
|