fix google download

teticio · Feb 10, 2024 · 7eddd7e · 7eddd7e
1 parent 5dca609
commit 7eddd7e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/download.py b/scripts/download.py
@@ -25,9 +25,9 @@ def download_file_from_google_drive(file_id, destination, size):
     params = {'export': 'dowloand', 'id': file_id, 'confirm': 't'}
     response = session.get(url, params=params, stream=True)
 
-    soup = BeautifulSoup(response.text, 'html.parser')
-    form = soup.find('form')
-    if form:
+    if 'text/html' in response.headers.get('Content-Type'):
+        soup = BeautifulSoup(response.text, 'html.parser')
+        form = soup.find('form')
         url = form['action']
         inputs = form.find_all('input')
         params = {input['name']: input.get('value', '') for input in inputs if input.get('name')}