Update image_query_classification.py

doppeltilde · Jun 12, 2024 · 8513221 · 8513221
1 parent 8855311
commit 8513221
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/routes/api/image_query_classification.py b/src/routes/api/image_query_classification.py
@@ -196,10 +196,10 @@ async def multi_image_query_classification(
         # Read the file as bytes
         image_list = []
 
-        classifier = check_model(model_name)
-
         for model_name in model_names:
             try:
+                classifier = check_model(model_name)
+
                 contents = await file.read()
 
                 labels_copy = labels.copy()