Upload folder using huggingface_hub

app.py

@@ -5,6 +5,9 @@ from huggingface_hub import hf_hub_download, HfApi
 from convert_to_quant import quantize
 
 @spaces.GPU(duration=300)
+def do_quantize(quant_args):
+    quantize(**quant_args)
+
 def run_quantization(
     source_repo,
     source_file,
@@ -74,7 +77,7 @@ def run_quantization(
         yield f"Quantizing to {output_filename}...\nThis may take a few minutes.", ""
 
         # We need to print output or capture it, convert_to_quant probably prints. We'll just run it.
-        quantize(**quant_args)
+        do_quantize(quant_args)
 
         yield f"Uploading {output_filename} to {target_repo}...", ""