elliesleightholm commited on
Commit
ab0c57f
·
1 Parent(s): fe26b6e

updating gpu

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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}")