Added tmp/images to store images

app.py

@@ -11,7 +11,6 @@ import requests
 from typing import Dict, List, Tuple, Union, Optional
 
 # dotenv.load_dotenv()
-FAL_KEY = os.getenv("FAL_KEY")
 
 # Configure logging
 logging.basicConfig(
@@ -55,6 +54,8 @@ CONFIG_FILE = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
 GROUNDINGDINO_CHECKPOINT = "groundingdino_swint_ogc.pth"
 SAM_CHECKPOINT = 'sam_hq_vit_l.pth'
 OUTPUT_DIR = "outputs"
+FAL_KEY = os.getenv("FAL_KEY")
+UPLOAD_DIR = "/tmp/images"
 
 # Global variables for model caching
 _models = {
@@ -326,8 +327,17 @@ def run_grounded_sam(
         else:
             return [Image.new('RGB', (400, 300), color='gray'), Image.new('RGBA', (400, 300), color=(0, 0, 0, 0))]
 
+def upload_file(file: UploadFile):
+    file_path = os.path.join(UPLOAD_DIR, file.filename)
+    with open(file_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+
+    # Return URL
+    return {"url": f"https://huggingface.co/spaces/rootglitch/CarVizGradioDemo01/{file_path}"}
+
 def generate_ai_bg(input_img, prompt):
-    hf_input_img = 'https://huggingface.co/spaces/rootglitch/CarVizGradioDemo01/'+input_img
+    hf_input_img = upload_file(input_img)["url"]
+
     handler = fal_client.submit(
         "fal-ai/iclight-v2",
         arguments={