mrcuddle commited on
Commit
c282e5d
·
verified ·
1 Parent(s): 03295cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import gradio as gr
2
  import subprocess
3
  from huggingface_hub import HfApi
 
4
 
 
5
  def merge_and_upload(weight_drop_prob, scaling_factor, base_model, model_to_merge, output_path, repo_name, token):
6
  # Construct the command to run hf_merge.py
7
  command = [
 
1
  import gradio as gr
2
  import subprocess
3
  from huggingface_hub import HfApi
4
+ import spaces
5
 
6
+ @spaces.GPU
7
  def merge_and_upload(weight_drop_prob, scaling_factor, base_model, model_to_merge, output_path, repo_name, token):
8
  # Construct the command to run hf_merge.py
9
  command = [