chore: download onnx-data on spaces

Signed-off-by: Suvaditya Mukherjee <suvadityamuk@gmail.com>

app.py

@@ -6,6 +6,7 @@ import requests
 import wandb
 import torch
 import spaces
+import tqdm
 import psutil
 import pymupdf
 import gradio as gr
@@ -101,9 +102,23 @@ if __name__ == "__main__":
     RESUME_URL = "https://drive.google.com/file/d/1YMF9NNTG5gubwJ7ipI5JfxAJKhlD9h2v/"
 
     ONNX_MODEL_PATH = "https://huggingface.co/onnx-community/Qwen2.5-1.5B-Instruct/resolve/main/onnx/model.onnx_data"
+    SAVE_PATH = "./model.onnx_data"
 
     print("Downloading ONNX model...")
     response = requests.get(ONNX_MODEL_PATH, stream=True)
+    response.raise_for_status()
+    
+    total_size = int(response.headers.get('content-length', 0))
+    
+    with open(SAVE_PATH, 'wb') as file, tqdm(
+        desc=os.path.basename(SAVE_PATH),
+        total=total_size,
+        unit='iB',
+        unit_scale=True
+    ) as pbar:
+        for data in response.iter_content(chunk_size=8192):
+            size = file.write(data)
+            pbar.update(size)
     print("Downloaded ONNX model!")
 
     # Download file