Commit
·
ab0c57f
1
Parent(s):
fe26b6e
updating gpu
Browse files
app.py
CHANGED
@@ -47,6 +47,7 @@ If you use Marqo-Ecommerce-L or Marqo-Ecommerce-B, please cite us:
|
|
47 |
"""
|
48 |
|
49 |
# Function to initialize a model, preprocess, and text features
|
|
|
50 |
def initialize_model(model_name, progress=gr.Progress()):
|
51 |
progress(0, f"Initializing model: {model_name}...")
|
52 |
model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(f"hf-hub:Marqo/{model_name}")
|
|
|
47 |
"""
|
48 |
|
49 |
# Function to initialize a model, preprocess, and text features
|
50 |
+
@spaces.GPU
|
51 |
def initialize_model(model_name, progress=gr.Progress()):
|
52 |
progress(0, f"Initializing model: {model_name}...")
|
53 |
model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(f"hf-hub:Marqo/{model_name}")
|