Automatically delete after 10 minutes.

app.py

@@ -1,13 +1,13 @@
-import torch  
 from meshgpt_pytorch import ( 
     MeshTransformer,
     mesh_render
 )
 import igl
-
 import gradio as gr
-import numpy as np 
 import tempfile
+import os
+import threading
+import time
 
 transformer = MeshTransformer.from_pretrained("MarcusLoren/MeshGPT-preview")
 
@@ -20,6 +20,11 @@ def save_as_obj(file_path):
     return file_path
 
 
+def delete_file_after_ten_minutes(filename):
+    time.sleep(600)  # Wait for 10 minutes
+    os.remove(filename)
+
+
 def predict(text, num_input, num_temp):
     transformer.eval()
     labels = [label.strip() for label in text.split(',')] 
@@ -32,7 +37,10 @@ def predict(text, num_input, num_temp):
 
     with tempfile.NamedTemporaryFile(suffix=".obj", delete=False) as temp_file:
         mesh_render.save_rendering(temp_file.name, output)
-        return save_as_obj(temp_file.name)
+        result = save_as_obj(temp_file.name)
+    threading.Thread(target=delete_file_after_ten_minutes, args=(temp_file.name,)).start()
+
+    return result
 
 gradio_app = gr.Interface(
     predict,