fix

app.py

@@ -67,8 +67,8 @@ async def upload_resize_image_url(session, image_url):
                 s3.upload_fileobj(Fileobj=image_bytes, Bucket=AWS_S3_BUCKET_NAME, Key="diffusers-gallery/" + fname,
                                 ExtraArgs={"ContentType": "image/jpeg", "CacheControl": "max-age=31536000"})
                 return fname
-
     except Exception as e:
+        print(f"Error uploading image {image_url}: {e}")
         return None
 
 
@@ -116,7 +116,6 @@ def run_classifier(images):
 
         # data response is array data:[[{img0}, {img1}, {img2}...], Label, Gallery],
         class_data = response['data'][0][0]
-        print(class_data)
         class_data_parsed = {row['label']: round(
             row['score'], 3) for row in class_data}
 
@@ -171,8 +170,9 @@ async def sync_data():
         downloads = model['downloads']
         model_card = fetch_model_card(model_id)
         images = await find_image_in_model_card(model_card)
-        images = [i for i in images if i is not None]
+
         classifier = run_classifier(images)
+        print(images, classifier)
         # update model row with image and classifier data
         with database.get_db() as db:
             cursor = db.cursor()
@@ -194,24 +194,21 @@ async def sync_data():
         models_no_images =list(cursor.fetchall())
         for model in tqdm(models_no_images):
             model_id = model['id']
+            model_data = json.loads(model['data'])
             print("Updating model", model_id)
             model_card = fetch_model_card(model_id)
             images = await find_image_in_model_card(model_card)
-            images = [i for i in images if i is not None]
-            print(images)
             classifier = run_classifier(images)
+
             # update model row with image and classifier data
             with database.get_db() as db:
                 cursor = db.cursor()
-                cursor.execute("UPDATE models SET data = json_set(data, '$.images', ?) WHERE id = ?",
-                               [json.dumps(images), model_id])
-                cursor.execute("UPDATE models SET data = json_set(data, '$.class', ?) WHERE id = ?",
-                               [json.dumps(classifier), model_id])
+                cursor.execute("UPDATE models SET data = ? WHERE id = ?",
+                               [json.dumps(model_data), model_id])
                 db.commit()
 
 
 
-
     print("Update likes and downloads")
     for model in tqdm(all_models):
         model_id = model['id']
@@ -224,8 +221,8 @@ async def sync_data():
             db.commit()
 
     print("Updating DB repository")
-    subprocess.Popen(
-        "git add . && git commit --amend -m 'update' && git push --force", cwd=DB_FOLDER, shell=True)
+    # subprocess.Popen(
+    #     "git add . && git commit --amend -m 'update' && git push --force", cwd=DB_FOLDER, shell=True)
 
 
 app = FastAPI()
@@ -300,7 +297,6 @@ def get_page(page: int = 1, sort: Sort = Sort.trending, style: Style = Style.all
         total_pages = (total + MAX_PAGE_SIZE - 1) // MAX_PAGE_SIZE
         models_data = []
         for result in results:
-            print(list(result))
             data = json.loads(result['data'])
             # update downloads and likes from db table
             data['downloads'] = result['downloads']