Update image_query_classification.py

doppeltilde · Jun 12, 2024 · 629fa39 · 629fa39
1 parent a402a25
commit 629fa39
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/routes/api/image_query_classification.py b/src/routes/api/image_query_classification.py
@@ -255,16 +255,14 @@ async def multi_image_query_classification(
                         )
                     finally:
                         img.close()
+                        del classifier
+                        torch.cuda.empty_cache()
 
             except Exception as e:
                 print("File is not a valid image.")
                 img.close()
                 return {"error": str(e)}
 
-            finally:
-                del classifier
-                torch.cuda.empty_cache()
-
         totalResults.append({index: image_list})
 
     return totalResults