🔧

files_cells/notebooks/en/downloading_en.ipynb

@@ -371,9 +371,10 @@
         "        model_name = data['files'][0]['name']\n",
         "\n",
         "    # Finding a safe image: less than level 4 | Kaggle\n",
-        "    if env == 'Kaggle':\n",
-        "        image_url = next((image['url'] for image in data['images'] if image['nsfwLevel'] < 4), None)\n",
-        "    else:\n",
+        "    image_url = ''\n",
+        "    if env == 'Kaggle' and data['images']:\n",
+        "        image_url = next((img['url'] for img in data['images'] if img['nsfwLevel'] < 4), None)\n",
+        "    elif data['images']:\n",
         "        image_url = data['images'][0]['url']\n",
         "\n",
         "    return model_type, model_name, image_url\n",

files_cells/notebooks/ru/downloading_ru.ipynb

@@ -371,9 +371,10 @@
         "        model_name = data['files'][0]['name']\n",
         "\n",
         "    # Finding a safe image: less than level 4 | Kaggle\n",
-        "    if env == 'Kaggle':\n",
-        "        image_url = next((image['url'] for image in data['images'] if image['nsfwLevel'] < 4), None)\n",
-        "    else:\n",
+        "    image_url = ''\n",
+        "    if env == 'Kaggle' and data['images']:\n",
+        "        image_url = next((img['url'] for img in data['images'] if img['nsfwLevel'] < 4), None)\n",
+        "    elif data['images']:\n",
         "        image_url = data['images'][0]['url']\n",
         "\n",
         "    return model_type, model_name, image_url\n",

files_cells/python/en/downloading_en.py

@@ -361,9 +361,10 @@ def extract_model_info(data, url):
         model_name = data['files'][0]['name']
 
     # Finding a safe image: less than level 4 | Kaggle
-    if env == 'Kaggle':
-        image_url = next((image['url'] for image in data['images'] if image['nsfwLevel'] < 4), None)
-    else:
+    image_url = ''
+    if env == 'Kaggle' and data['images']:
+        image_url = next((img['url'] for img in data['images'] if img['nsfwLevel'] < 4), None)
+    elif data['images']:
         image_url = data['images'][0]['url']
 
     return model_type, model_name, image_url

files_cells/python/ru/downloading_ru.py

@@ -361,9 +361,10 @@ def extract_model_info(data, url):
         model_name = data['files'][0]['name']
 
     # Finding a safe image: less than level 4 | Kaggle
-    if env == 'Kaggle':
-        image_url = next((image['url'] for image in data['images'] if image['nsfwLevel'] < 4), None)
-    else:
+    image_url = ''
+    if env == 'Kaggle' and data['images']:
+        image_url = next((img['url'] for img in data['images'] if img['nsfwLevel'] < 4), None)
+    elif data['images']:
         image_url = data['images'][0]['url']
 
     return model_type, model_name, image_url