Update app.py

app.py

@@ -1,9 +1,7 @@
 import gradio as gr
 import subprocess
-from huggingface_hub import HfApi, HfFolder
-import spaces
+from huggingface_hub import HfApi
 
-@spaces.GPU
 def merge_and_upload(weight_drop_prob, scaling_factor, base_model, model_to_merge, output_path, repo_name, token):
     # Construct the command to run hf_merge.py
     command = [
@@ -41,15 +39,15 @@ def merge_and_upload(weight_drop_prob, scaling_factor, base_model, model_to_merg
 iface = gr.Interface(
     fn=merge_and_upload,
     inputs=[
-        gr.inputs.Slider(minimum=0, maximum=1, default=0.13, label="Weight Drop Probability"),
-        gr.inputs.Number(default=3.0, label="Scaling Factor"),
-        gr.inputs.Textbox(label="Base Model File/Folder"),
-        gr.inputs.Textbox(label="Model to Merge"),
-        gr.inputs.Textbox(label="Output Path"),
-        gr.inputs.Textbox(label="Hugging Face Repo Name"),
-        gr.inputs.Textbox(label="Hugging Face Token", type="password")
+        gr.Slider(minimum=0, maximum=1, value=0.13, label="Weight Drop Probability"),
+        gr.Number(value=3.0, label="Scaling Factor"),
+        gr.Textbox(label="Base Model File/Folder"),
+        gr.Textbox(label="Model to Merge"),
+        gr.Textbox(label="Output Path"),
+        gr.Textbox(label="Hugging Face Repo Name"),
+        gr.Textbox(label="Hugging Face Token", type="password")
     ],
-    outputs="text",
+    outputs=gr.Textbox(label="Output"),
     title="Model Merger and Uploader",
     description="Merge two models using the Super Mario merge method and upload to Hugging Face Hub."
 )