Update app.py

app.py

@@ -5,12 +5,10 @@ import time
 import os
 import shutil
 from gradio_client import Client, handle_file
-
-# Ensure /data/ directory exists
-os.makedirs('/data', exist_ok=True)
+# import spaces
 
 # Database setup
-conn = sqlite3.connect('/data/jobs.db', check_same_thread=False)
+conn = sqlite3.connect('/tmp/jobs.db', check_same_thread=False)
 c = conn.cursor()
 c.execute('''CREATE TABLE IF NOT EXISTS jobs
              (id INTEGER PRIMARY KEY, image_path TEXT, job_id TEXT, status TEXT, output_path TEXT)''')
@@ -19,6 +17,8 @@ conn.commit()
 # TRELLIS API client
 trellis_client = Client("jkorstad/TRELLIS")
 
+
+# @spaces.GPU
 # Processing logic with three-step TRELLIS workflow
 def process_job(job_id):
     try:
@@ -63,8 +63,8 @@ def process_job(job_id):
         glb_path = glb_result[0]  # First element is the GLB filepath
         
         # Move GLB to persistent storage
-        output_path = f'/data/outputs/result_{job_id}.glb'
-        os.makedirs('/data/outputs', exist_ok=True)
+        output_path = f'/tmp/outputs/result_{job_id}.glb'
+        os.makedirs('/tmp/outputs', exist_ok=True)
         shutil.move(glb_path, output_path)
         
         # Update job status
@@ -84,8 +84,8 @@ def submit_images(files):
         c.execute("INSERT INTO jobs (status) VALUES ('submitted')")
         job_id = c.lastrowid
         conn.commit()
-        image_path = f'/data/inputs/input_{job_id}.jpg'
-        os.makedirs('/data/inputs', exist_ok=True)
+        image_path = f'/tmp/inputs/input_{job_id}.jpg'
+        os.makedirs('/tmp/inputs', exist_ok=True)
         shutil.copy(file.name, image_path)
         c.execute("UPDATE jobs SET image_path=? WHERE id=?", (image_path, job_id))
         conn.commit()