Update classification.py
Browse files- classification.py +0 -1
classification.py
CHANGED
|
@@ -133,7 +133,6 @@ def load_data(file_obj):
|
|
| 133 |
# Assuming file_obj is a file-like object uploaded via Gradio, use `pd.read_excel` directly on it
|
| 134 |
return pd.read_excel(file_obj)
|
| 135 |
|
| 136 |
-
@spaces.GPU
|
| 137 |
def initialize_models():
|
| 138 |
model_ST = SentenceTransformer("all-mpnet-base-v2",device = "cuda")
|
| 139 |
return model_ST
|
|
|
|
| 133 |
# Assuming file_obj is a file-like object uploaded via Gradio, use `pd.read_excel` directly on it
|
| 134 |
return pd.read_excel(file_obj)
|
| 135 |
|
|
|
|
| 136 |
def initialize_models():
|
| 137 |
model_ST = SentenceTransformer("all-mpnet-base-v2",device = "cuda")
|
| 138 |
return model_ST
|