Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import subprocess
+import spaces
 
+@spaces.GPU
 def merge_models(weight_drop_prob, scaling_factor, base_model, model_to_merge, output_path):
     # Construct the command to run hf_merge.py
     command = [