Update video_classification.py

doppeltilde · Jun 12, 2024 · 4ec45a1 · 4ec45a1
1 parent 43922c4
commit 4ec45a1
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/routes/api/video_classification.py b/src/routes/api/video_classification.py
@@ -122,14 +122,14 @@ async def video_classification(
             except Exception as e:
                 return {"error": e}
 
+            finally:
+                tf.close()
+                os.remove(tf.name)
+                del classifier
+                torch.cuda.empty_cache()
+
         return totalResults
 
     except Exception as e:
         print("File is not a valid image.")
         return {"error": str(e)}
-
-    finally:
-        tf.close()
-        os.remove(tf.name)
-        del classifier
-        torch.cuda.empty_cache()