suvadityamuk commited on
Commit
1514f66
·
1 Parent(s): 7572f69

chore: download onnx-data on spaces

Browse files

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

Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import os
2
  import re
3
  import json
4
  import time
 
5
  import wandb
6
  import torch
7
  import spaces
@@ -99,6 +100,10 @@ if __name__ == "__main__":
99
  RESUME_PATH = os.path.join(os.getcwd(), "Resume.pdf")
100
  RESUME_URL = "https://drive.google.com/file/d/1YMF9NNTG5gubwJ7ipI5JfxAJKhlD9h2v/"
101
 
 
 
 
 
102
  # Download file
103
  download_pdf_from_gdrive(RESUME_URL, RESUME_PATH)
104
 
 
2
  import re
3
  import json
4
  import time
5
+ import requests
6
  import wandb
7
  import torch
8
  import spaces
 
100
  RESUME_PATH = os.path.join(os.getcwd(), "Resume.pdf")
101
  RESUME_URL = "https://drive.google.com/file/d/1YMF9NNTG5gubwJ7ipI5JfxAJKhlD9h2v/"
102
 
103
+ ONNX_MODEL_PATH = "https://huggingface.co/onnx-community/Qwen2.5-1.5B-Instruct/resolve/main/onnx/model.onnx_data"
104
+
105
+ response = requests.get(ONNX_MODEL_PATH, stream=True)
106
+
107
  # Download file
108
  download_pdf_from_gdrive(RESUME_URL, RESUME_PATH)
109