Download from GH release instead of GDrive

app.py

@@ -2,19 +2,23 @@ from PIL import Image
 from model import yolox
 from os import listdir
 import os.path
-import gdown
+import requests
 import gradio as gr
 import numpy as np
 import streamlit as st
 
-DMINITY_MODEL_URL = "https://drive.google.com/uc?id=1gb3Pq5UM41IvkqQqe4vDExSBza9dmwr0"
+DMINITY_MODEL_URL = "https://github.com/Dolpheyn/dminity/releases/download/v-1.0.0/dminity.onnx"
 MODEL_PATH = "dminity.onnx"
 
 @st.cache(allow_output_mutation=True, show_spinner=True)
 def get_model():
     # Download model from Google Drive if it does not exist
     if not os.path.isfile(MODEL_PATH):
-        gdown.download(DMINITY_MODEL_URL, MODEL_PATH)
+        print("Downloading dminity model weight from {}...".format(DMINITY_MODEL_URL))
+        r = requests.get(DMINITY_MODEL_URL, allow_redirects=True)        
+        print("Writing to {}".format(MODEL_PATH))
+        open(MODEL_PATH, 'wb').write(r.content)        
+        print("Done!")
     
     # Load model with OpenCV
     model = yolox(MODEL_PATH, p6=False, confThreshold=0.3)