Update app.py (#2)

- Update app.py (fb491a0bdb50313633d8674d451345e0c0deb38a)


Co-authored-by: Muhammad Mudassir <Mudassir-75@users.noreply.huggingface.co>

app.py

@@ -7,6 +7,7 @@ import torch
 import os
 import requests
 from tempfile import NamedTemporaryFile
+import gc
 
 
 import tensorflow_hub as hub
@@ -58,6 +59,9 @@ app = Flask(__name__)
 @app.route('/process_video', methods=['POST'])
 def process_video():
     try:
+        # Clear previous cache
+        gc.collect()
+        torch.cuda.empty_cache()
         # Get the video URL from the request
         video_url = request.json.get('videoURL')
         height =  request.json.get('height')
@@ -278,5 +282,13 @@ def process_video():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+# if __name__ == '__main__':
+#     app.run(host='0.0.0.0', port=7860)
+
 if __name__ == '__main__':
+    # Clear any cache before starting the Flask server
+    gc.collect()
+    torch.cuda.empty_cache()
+    
+    # Start the Flask app
     app.run(host='0.0.0.0', port=7860)