optimize

app.py

@@ -55,71 +55,70 @@ class Preprocessor:
     def __call__(self, image: Image.Image, **kwargs) -> Image.Image:
         return self.model(image, **kwargs)
 
-# torch.cuda.max_memory_allocated(device="cuda")
+if gr.NO_RELOAD:
+    # Controlnet Normal
+    model_id = "lllyasviel/control_v11p_sd15_normalbae"
+    print("initializing controlnet")
+    controlnet = ControlNetModel.from_pretrained(
+        model_id,
+        torch_dtype=torch.float16,
+        attn_implementation="flash_attention_2",
+    ).to("cuda")
 
-# Controlnet Normal
-model_id = "lllyasviel/control_v11p_sd15_normalbae"
-print("initializing controlnet")
-controlnet = ControlNetModel.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16,
-    attn_implementation="flash_attention_2",
-).to("cuda")
-
-# Scheduler
-scheduler = DPMSolverMultistepScheduler.from_pretrained(
-    "runwayml/stable-diffusion-v1-5",
-    solver_order=2,
-    subfolder="scheduler",
-    use_karras_sigmas=True,
-    final_sigmas_type="sigma_min",
-    algorithm_type="sde-dpmsolver++",
-    prediction_type="epsilon",
-    thresholding=False,
-    denoise_final=True,
-    device_map="cuda",
-    torch_dtype=torch.float16,
-)
+    # Scheduler
+    scheduler = DPMSolverMultistepScheduler.from_pretrained(
+        "runwayml/stable-diffusion-v1-5",
+        solver_order=2,
+        subfolder="scheduler",
+        use_karras_sigmas=True,
+        final_sigmas_type="sigma_min",
+        algorithm_type="sde-dpmsolver++",
+        prediction_type="epsilon",
+        thresholding=False,
+        denoise_final=True,
+        device_map="cuda",
+        torch_dtype=torch.float16,
+    )
 
-# Stable Diffusion Pipeline URL
-# base_model_url = "https://huggingface.co/broyang/hentaidigitalart_v20/blob/main/realcartoon3d_v15.safetensors"
-base_model_url = "https://huggingface.co/Lykon/AbsoluteReality/blob/main/AbsoluteReality_1.8.1_pruned.safetensors"
-# vae_url = "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/vae-ft-mse-840000-ema-pruned.safetensors"
+    # Stable Diffusion Pipeline URL
+    # base_model_url = "https://huggingface.co/broyang/hentaidigitalart_v20/blob/main/realcartoon3d_v15.safetensors"
+    base_model_url = "https://huggingface.co/Lykon/AbsoluteReality/blob/main/AbsoluteReality_1.8.1_pruned.safetensors"
+    # vae_url = "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/vae-ft-mse-840000-ema-pruned.safetensors"
 
-# print('loading vae')
-# vae = AutoencoderKL.from_single_file(vae_url, torch_dtype=torch.float16).to("cuda")
-# vae.to(memory_format=torch.channels_last) 
+    # print('loading vae')
+    # vae = AutoencoderKL.from_single_file(vae_url, torch_dtype=torch.float16).to("cuda")
+    # vae.to(memory_format=torch.channels_last) 
 
-print('loading pipe')
-pipe = StableDiffusionControlNetPipeline.from_single_file(
-    base_model_url,
-    safety_checker=None,
-    controlnet=controlnet,
-    scheduler=scheduler,
-    # vae=vae,
-    torch_dtype=torch.float16,
-).to("cuda")
+    print('loading pipe')
+    pipe = StableDiffusionControlNetPipeline.from_single_file(
+        base_model_url,
+        safety_checker=None,
+        controlnet=controlnet,
+        scheduler=scheduler,
+        # vae=vae,
+        torch_dtype=torch.float16,
+    ).to("cuda")
 
-print("loading preprocessor")
-preprocessor = Preprocessor()
-preprocessor.load("NormalBae")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="EasyNegativeV2.safetensors", token="EasyNegativeV2",)
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="badhandv4.pt", token="badhandv4")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="fcNeg-neg.pt", token="fcNeg-neg")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Ahegao.pt", token="HDA_Ahegao")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Bondage.pt", token="HDA_Bondage")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_pet_play.pt", token="HDA_pet_play")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_unconventional maid.pt", token="HDA_unconventional_maid")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NakedHoodie.pt", token="HDA_NakedHoodie")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NunDress.pt", token="HDA_NunDress")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Shibari.pt", token="HDA_Shibari")
-pipe.to("cuda")
+    print("loading preprocessor")
+    preprocessor = Preprocessor()
+    preprocessor.load("NormalBae")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="EasyNegativeV2.safetensors", token="EasyNegativeV2",)
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="badhandv4.pt", token="badhandv4")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="fcNeg-neg.pt", token="fcNeg-neg")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Ahegao.pt", token="HDA_Ahegao")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Bondage.pt", token="HDA_Bondage")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_pet_play.pt", token="HDA_pet_play")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_unconventional maid.pt", token="HDA_unconventional_maid")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NakedHoodie.pt", token="HDA_NakedHoodie")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NunDress.pt", token="HDA_NunDress")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Shibari.pt", token="HDA_Shibari")
+    pipe.to("cuda")
 
-print("---------------Loaded controlnet pipeline---------------") 
-torch.cuda.empty_cache()
-gc.collect()
-print(f"CUDA memory allocated: {torch.cuda.max_memory_allocated(device='cuda') / 1e9:.2f} GB")
-print("Model Compiled!")
+    print("---------------Loaded controlnet pipeline---------------") 
+    torch.cuda.empty_cache()
+    gc.collect()
+    print(f"CUDA memory allocated: {torch.cuda.max_memory_allocated(device='cuda') / 1e9:.2f} GB")
+    print("Model Compiled!")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
@@ -135,7 +134,7 @@ def get_additional_prompt():
     # outfit = ["schoolgirl outfit", "playboy outfit", "red dress", "gala dress", "cheerleader outfit", "nurse outfit", "Kimono"]
 
 def get_prompt(prompt, additional_prompt):
-    interior = "design-style interior designed (interior space), captured with a DSLR camera using f/10 aperture, 1/60 sec shutter speed, ISO 400, 20mm focal length, tungsten white balance"
+    interior = "design-style interior designed (interior space),tungsten white balance,captured with a DSLR camera using f/10 aperture, 1/60 sec shutter speed, ISO 400, 20mm focal length"
     default = "hyperrealistic photography,extremely detailed,(intricate details),unity 8k wallpaper,ultra detailed"
     default2 = f"professional 3d model {prompt},octane render,highly detailed,volumetric,dramatic lighting,hyperrealistic photography,extremely detailed,(intricate details),unity 8k wallpaper,ultra detailed"
     randomize = get_additional_prompt()
@@ -173,51 +172,51 @@ style_list = [
     },
     {
         "name": "Minimalistic",
-        "prompt": "Minimalist interior design, clean lines, neutral colors, uncluttered space, functional furniture, lots of natural light"
+        "prompt": "Minimalist interior design,clean lines,neutral colors,uncluttered space,functional furniture,lots of natural light"
     },
     {
         "name": "Boho",
-        "prompt": "Bohemian chic interior, eclectic mix of patterns and textures, vintage furniture, plants, woven textiles, warm earthy colors"
+        "prompt": "Bohemian chic interior,eclectic mix of patterns and textures,vintage furniture,plants,woven textiles,warm earthy colors"
     },
     {
         "name": "Farmhouse",
-        "prompt": "Modern farmhouse interior, rustic wood elements, shiplap walls, neutral color palette, industrial accents, cozy textiles"
+        "prompt": "Modern farmhouse interior,rustic wood elements,shiplap walls,neutral color palette,industrial accents,cozy textiles"
     },
     {
         "name": "Saudi Prince",
-        "prompt": "Opulent gold interior, luxurious ornate furniture, crystal chandeliers, rich fabrics, marble floors, intricate Arabic patterns"
+        "prompt": "Opulent gold interior,luxurious ornate furniture,crystal chandeliers,rich fabrics,marble floors,intricate Arabic patterns"
     },
     {
         "name": "Neoclassical",
-        "prompt": "Neoclassical interior design, elegant columns, ornate moldings, symmetrical layout, refined furniture, muted color palette"
+        "prompt": "Neoclassical interior design,elegant columns,ornate moldings,symmetrical layout,refined furniture,muted color palette"
     },
     {
         "name": "Eclectic",
-        "prompt": "Eclectic interior design, mix of styles and eras, bold color combinations, diverse furniture pieces, unique art objects"
+        "prompt": "Eclectic interior design,mix of styles and eras,bold color combinations,diverse furniture pieces,unique art objects"
     },
     {
         "name": "Parisian",
-        "prompt": "Parisian apartment interior, all-white color scheme, ornate moldings, herringbone wood floors, elegant furniture, large windows"
+        "prompt": "Parisian apartment interior,all-white color scheme,ornate moldings,herringbone wood floors,elegant furniture,large windows"
     },
     {
         "name": "Hollywood",
-        "prompt": "Hollywood Regency interior, glamorous and luxurious, bold colors, mirrored surfaces, velvet upholstery, gold accents"
+        "prompt": "Hollywood Regency interior,glamorous and luxurious,bold colors,mirrored surfaces,velvet upholstery,gold accents"
     },
     {
         "name": "Scandinavian",
-        "prompt": "Scandinavian interior design, light wood tones, white walls, minimalist furniture, cozy textiles, hygge atmosphere"
+        "prompt": "Scandinavian interior design,light wood tones,white walls,minimalist furniture,cozy textiles,hygge atmosphere"
     },
     {
         "name": "Beach",
-        "prompt": "Coastal beach house interior, light blue and white color scheme, weathered wood, nautical accents, sheer curtains, ocean view"
+        "prompt": "Coastal beach house interior,light blue and white color scheme,weathered wood,nautical accents,sheer curtains,ocean view"
     },
     {
         "name": "Japanese",
-        "prompt": "Traditional Japanese interior, tatami mats, shoji screens, low furniture, zen garden view, minimalist decor, natural materials"
+        "prompt": "Traditional Japanese interior,tatami mats,shoji screens,low furniture,zen garden view,minimalist decor,natural materials"
     },
     { 
         "name": "Midcentury Modern",
-        "prompt": "Mid-century modern interior, 1950s-60s style furniture, organic shapes, warm wood tones, bold accent colors, large windows"
+        "prompt": "Mid-century modern interior,1950s-60s style furniture,organic shapes,warm wood tones,bold accent colors,large windows"
     },
     {
         "name": "Retro Futurism",
@@ -225,11 +224,11 @@ style_list = [
     },
     {
         "name": "Texan",
-        "prompt": "Western cowboy interior, rustic wood beams, leather furniture, cowhide rugs, antler chandeliers, southwestern patterns"
+        "prompt": "Western cowboy interior,rustic wood beams,leather furniture,cowhide rugs,antler chandeliers,southwestern patterns"
     },
     {
         "name": "Matrix",
-        "prompt": "Futuristic cyberpunk interior, neon accent lighting, holographic plants, sleek black surfaces, advanced gaming setup, transparent screens, Blade Runner inspired decor, high-tech minimalist furniture"
+        "prompt": "Futuristic cyberpunk interior,neon accent lighting,holographic plants,sleek black surfaces,advanced gaming setup,transparent screens,Blade Runner inspired decor,high-tech minimalist furniture"
     }
 ] 
 
@@ -238,7 +237,7 @@ STYLE_NAMES = list(styles.keys())
 
 def apply_style(style_name):
     if style_name in styles:
-        p = styles.get(style_name, "boho chic")
+        p = styles.get(style_name, "none")
     return p
 
     
@@ -416,7 +415,7 @@ def process_image(
     )
     preprocess_time = time.time() - preprocess_start
     if style_selection is not None or style_selection != "None":
-        prompt = "Photo from Pinterest of " + apply_style(style_selection) + " " + prompt + " " + a_prompt
+        prompt = "Photo from Pinterest of " + apply_style(style_selection) + " " + prompt
     else:
         prompt=str(get_prompt(prompt, a_prompt))
     negative_prompt=str(n_prompt)

local_app.py

@@ -11,10 +11,11 @@ import time
 import gradio as gr
 import numpy as np
 # import spaces
-# import imageio
+import imageio
 import gc
 import torch
 from PIL import Image
+from huggingface_hub import HfApi
 from diffusers import (
     ControlNetModel,
     DPMSolverMultistepScheduler,
@@ -25,6 +26,7 @@ from controlnet_aux_local import NormalBaeDetector
 
 MAX_SEED = np.iinfo(np.int32).max
 API_KEY = os.environ.get("API_KEY", None)
+api = HfApi()
 # os.environ['HF_HOME'] = '/data/.huggingface'
 
 print("CUDA version:", torch.version.cuda)
@@ -53,71 +55,70 @@ class Preprocessor:
     def __call__(self, image: Image.Image, **kwargs) -> Image.Image:
         return self.model(image, **kwargs)
 
-# torch.cuda.max_memory_allocated(device="cuda")
+if gr.NO_RELOAD:
+    # Controlnet Normal
+    model_id = "lllyasviel/control_v11p_sd15_normalbae"
+    print("initializing controlnet")
+    controlnet = ControlNetModel.from_pretrained(
+        model_id,
+        torch_dtype=torch.float16,
+        attn_implementation="flash_attention_2",
+    ).to("cuda")
 
-# Controlnet Normal
-model_id = "lllyasviel/control_v11p_sd15_normalbae"
-print("initializing controlnet")
-controlnet = ControlNetModel.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16,
-    attn_implementation="flash_attention_2",
-).to("cuda")
-
-# Scheduler
-scheduler = DPMSolverMultistepScheduler.from_pretrained(
-    "runwayml/stable-diffusion-v1-5",
-    solver_order=2,
-    subfolder="scheduler",
-    use_karras_sigmas=True,
-    final_sigmas_type="sigma_min",
-    algorithm_type="sde-dpmsolver++",
-    prediction_type="epsilon",
-    thresholding=False,
-    denoise_final=True,
-    device_map="cuda",
-    torch_dtype=torch.float16,
-)
+    # Scheduler
+    scheduler = DPMSolverMultistepScheduler.from_pretrained(
+        "runwayml/stable-diffusion-v1-5",
+        solver_order=2,
+        subfolder="scheduler",
+        use_karras_sigmas=True,
+        final_sigmas_type="sigma_min",
+        algorithm_type="sde-dpmsolver++",
+        prediction_type="epsilon",
+        thresholding=False,
+        denoise_final=True,
+        device_map="cuda",
+        torch_dtype=torch.float16,
+    )
 
-# Stable Diffusion Pipeline URL
-# base_model_url = "https://huggingface.co/broyang/hentaidigitalart_v20/blob/main/realcartoon3d_v15.safetensors"
-base_model_url = "https://huggingface.co/Lykon/AbsoluteReality/blob/main/AbsoluteReality_1.8.1_pruned.safetensors"
-# vae_url = "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/vae-ft-mse-840000-ema-pruned.safetensors"
+    # Stable Diffusion Pipeline URL
+    # base_model_url = "https://huggingface.co/broyang/hentaidigitalart_v20/blob/main/realcartoon3d_v15.safetensors"
+    base_model_url = "https://huggingface.co/Lykon/AbsoluteReality/blob/main/AbsoluteReality_1.8.1_pruned.safetensors"
+    # vae_url = "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/vae-ft-mse-840000-ema-pruned.safetensors"
 
-# print('loading vae')
-# vae = AutoencoderKL.from_single_file(vae_url, torch_dtype=torch.float16).to("cuda")
-# vae.to(memory_format=torch.channels_last) 
+    # print('loading vae')
+    # vae = AutoencoderKL.from_single_file(vae_url, torch_dtype=torch.float16).to("cuda")
+    # vae.to(memory_format=torch.channels_last) 
 
-print('loading pipe')
-pipe = StableDiffusionControlNetPipeline.from_single_file(
-    base_model_url,
-    safety_checker=None,
-    controlnet=controlnet,
-    scheduler=scheduler,
-    # vae=vae,
-    torch_dtype=torch.float16,
-).to("cuda")
+    print('loading pipe')
+    pipe = StableDiffusionControlNetPipeline.from_single_file(
+        base_model_url,
+        safety_checker=None,
+        controlnet=controlnet,
+        scheduler=scheduler,
+        # vae=vae,
+        torch_dtype=torch.float16,
+    ).to("cuda")
 
-print("loading preprocessor")
-preprocessor = Preprocessor()
-preprocessor.load("NormalBae")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="EasyNegativeV2.safetensors", token="EasyNegativeV2",)
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="badhandv4.pt", token="badhandv4")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="fcNeg-neg.pt", token="fcNeg-neg")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Ahegao.pt", token="HDA_Ahegao")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Bondage.pt", token="HDA_Bondage")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_pet_play.pt", token="HDA_pet_play")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_unconventional maid.pt", token="HDA_unconventional_maid")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NakedHoodie.pt", token="HDA_NakedHoodie")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NunDress.pt", token="HDA_NunDress")
-# pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Shibari.pt", token="HDA_Shibari")
-pipe.to("cuda")
+    print("loading preprocessor")
+    preprocessor = Preprocessor()
+    preprocessor.load("NormalBae")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="EasyNegativeV2.safetensors", token="EasyNegativeV2",)
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="badhandv4.pt", token="badhandv4")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="fcNeg-neg.pt", token="fcNeg-neg")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Ahegao.pt", token="HDA_Ahegao")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Bondage.pt", token="HDA_Bondage")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_pet_play.pt", token="HDA_pet_play")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_unconventional maid.pt", token="HDA_unconventional_maid")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NakedHoodie.pt", token="HDA_NakedHoodie")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_NunDress.pt", token="HDA_NunDress")
+    # pipe.load_textual_inversion("broyang/hentaidigitalart_v20", weight_name="HDA_Shibari.pt", token="HDA_Shibari")
+    pipe.to("cuda")
 
-print("---------------Loaded controlnet pipeline---------------") 
-torch.cuda.empty_cache()
-gc.collect()
-print(f"CUDA memory allocated: {torch.cuda.max_memory_allocated(device='cuda') / 1e9:.2f} GB")
-print("Model Compiled!")
+    print("---------------Loaded controlnet pipeline---------------") 
+    torch.cuda.empty_cache()
+    gc.collect()
+    print(f"CUDA memory allocated: {torch.cuda.max_memory_allocated(device='cuda') / 1e9:.2f} GB")
+    print("Model Compiled!")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
@@ -133,7 +134,7 @@ def get_additional_prompt():
     # outfit = ["schoolgirl outfit", "playboy outfit", "red dress", "gala dress", "cheerleader outfit", "nurse outfit", "Kimono"]
 
 def get_prompt(prompt, additional_prompt):
-    interior = "design-style interior designed (interior space), captured with a DSLR camera using f/10 aperture, 1/60 sec shutter speed, ISO 400, 20mm focal length, tungsten white balance"
+    interior = "design-style interior designed (interior space),tungsten white balance,captured with a DSLR camera using f/10 aperture, 1/60 sec shutter speed, ISO 400, 20mm focal length"
     default = "hyperrealistic photography,extremely detailed,(intricate details),unity 8k wallpaper,ultra detailed"
     default2 = f"professional 3d model {prompt},octane render,highly detailed,volumetric,dramatic lighting,hyperrealistic photography,extremely detailed,(intricate details),unity 8k wallpaper,ultra detailed"
     randomize = get_additional_prompt()
@@ -160,7 +161,7 @@ def get_prompt(prompt, additional_prompt):
         prompt = f"boho chic"
         # print(f"-------------{preset}-------------")
     else:
-        prompt = f"Photo from Pinterest of {prompt} {interior}"
+        prompt = f"Photo from Pinterest of {prompt}{interior}"
         # prompt = default2
     return f"{prompt} f{additional_prompt}"
 
@@ -171,51 +172,51 @@ style_list = [
     },
     {
         "name": "Minimalistic",
-        "prompt": "Minimalist interior design, clean lines, neutral colors, uncluttered space, functional furniture, lots of natural light"
+        "prompt": "Minimalist interior design,clean lines,neutral colors,uncluttered space,functional furniture,lots of natural light"
     },
     {
         "name": "Boho",
-        "prompt": "Bohemian chic interior, eclectic mix of patterns and textures, vintage furniture, plants, woven textiles, warm earthy colors"
+        "prompt": "Bohemian chic interior,eclectic mix of patterns and textures,vintage furniture,plants,woven textiles,warm earthy colors"
     },
     {
         "name": "Farmhouse",
-        "prompt": "Modern farmhouse interior, rustic wood elements, shiplap walls, neutral color palette, industrial accents, cozy textiles"
+        "prompt": "Modern farmhouse interior,rustic wood elements,shiplap walls,neutral color palette,industrial accents,cozy textiles"
     },
     {
         "name": "Saudi Prince",
-        "prompt": "Opulent gold interior, luxurious ornate furniture, crystal chandeliers, rich fabrics, marble floors, intricate Arabic patterns"
+        "prompt": "Opulent gold interior,luxurious ornate furniture,crystal chandeliers,rich fabrics,marble floors,intricate Arabic patterns"
     },
     {
         "name": "Neoclassical",
-        "prompt": "Neoclassical interior design, elegant columns, ornate moldings, symmetrical layout, refined furniture, muted color palette"
+        "prompt": "Neoclassical interior design,elegant columns,ornate moldings,symmetrical layout,refined furniture,muted color palette"
     },
     {
         "name": "Eclectic",
-        "prompt": "Eclectic interior design, mix of styles and eras, bold color combinations, diverse furniture pieces, unique art objects"
+        "prompt": "Eclectic interior design,mix of styles and eras,bold color combinations,diverse furniture pieces,unique art objects"
     },
     {
         "name": "Parisian",
-        "prompt": "Parisian apartment interior, all-white color scheme, ornate moldings, herringbone wood floors, elegant furniture, large windows"
+        "prompt": "Parisian apartment interior,all-white color scheme,ornate moldings,herringbone wood floors,elegant furniture,large windows"
     },
     {
         "name": "Hollywood",
-        "prompt": "Hollywood Regency interior, glamorous and luxurious, bold colors, mirrored surfaces, velvet upholstery, gold accents"
+        "prompt": "Hollywood Regency interior,glamorous and luxurious,bold colors,mirrored surfaces,velvet upholstery,gold accents"
     },
     {
         "name": "Scandinavian",
-        "prompt": "Scandinavian interior design, light wood tones, white walls, minimalist furniture, cozy textiles, hygge atmosphere"
+        "prompt": "Scandinavian interior design,light wood tones,white walls,minimalist furniture,cozy textiles,hygge atmosphere"
     },
     {
         "name": "Beach",
-        "prompt": "Coastal beach house interior, light blue and white color scheme, weathered wood, nautical accents, sheer curtains, ocean view"
+        "prompt": "Coastal beach house interior,light blue and white color scheme,weathered wood,nautical accents,sheer curtains,ocean view"
     },
     {
         "name": "Japanese",
-        "prompt": "Traditional Japanese interior, tatami mats, shoji screens, low furniture, zen garden view, minimalist decor, natural materials"
+        "prompt": "Traditional Japanese interior,tatami mats,shoji screens,low furniture,zen garden view,minimalist decor,natural materials"
     },
     { 
         "name": "Midcentury Modern",
-        "prompt": "Mid-century modern interior, 1950s-60s style furniture, organic shapes, warm wood tones, bold accent colors, large windows"
+        "prompt": "Mid-century modern interior,1950s-60s style furniture,organic shapes,warm wood tones,bold accent colors,large windows"
     },
     {
         "name": "Retro Futurism",
@@ -223,11 +224,11 @@ style_list = [
     },
     {
         "name": "Texan",
-        "prompt": "Western cowboy interior, rustic wood beams, leather furniture, cowhide rugs, antler chandeliers, southwestern patterns"
+        "prompt": "Western cowboy interior,rustic wood beams,leather furniture,cowhide rugs,antler chandeliers,southwestern patterns"
     },
     {
         "name": "Matrix",
-        "prompt": "Futuristic cyberpunk interior, neon accent lighting, holographic plants, sleek black surfaces, advanced gaming setup, transparent screens, Blade Runner inspired decor, high-tech minimalist furniture"
+        "prompt": "Futuristic cyberpunk interior,neon accent lighting,holographic plants,sleek black surfaces,advanced gaming setup,transparent screens,Blade Runner inspired decor,high-tech minimalist furniture"
     }
 ] 
 
@@ -236,7 +237,7 @@ STYLE_NAMES = list(styles.keys())
 
 def apply_style(style_name):
     if style_name in styles:
-        p = styles.get(style_name, "boho chic")
+        p = styles.get(style_name, "none")
     return p
 
     
@@ -415,9 +416,9 @@ def process_image(
     )
     preprocess_time = time.time() - preprocess_start
     if style_selection is not None or style_selection != "None":
-        prompt = "Photo from Pinterest of " + apply_style(style_selection) + " " + prompt + " " + a_prompt
+        prompt = "Photo from Pinterest of " + apply_style(style_selection) + " " + prompt
     else:
-        prompt=str(get_prompt(prompt, a_prompt))
+        prompt=str(get_prompt(prompt))
     negative_prompt=str(n_prompt)
     print(prompt)
     print(f"\n-------------------------Preprocess done in: {preprocess_time:.2f} seconds-------------------------")    
@@ -434,6 +435,28 @@ def process_image(
     print(f"\n-------------------------Inference done in: {time.time() - start:.2f} seconds-------------------------")
     # torch.cuda.synchronize()
     torch.cuda.empty_cache()
+    # # upload block
+    # timestamp = int(time.time())
+    # img_path = f"{timestamp}.jpg"
+    # results_path = f"{timestamp}_out.jpg"
+    # imageio.imsave(img_path, image)
+    # imageio.imsave(results_path, results)
+    # api.upload_file(
+    #     path_or_fileobj=img_path,
+    #     path_in_repo=img_path,
+    #     repo_id="broyang/interior-ai-outputs",
+    #     repo_type="dataset",
+    #     token=API_KEY,
+    #     run_as_future=True,
+    # )
+    # api.upload_file(
+    #     path_or_fileobj=results_path,
+    #     path_in_repo=results_path,
+    #     repo_id="broyang/interior-ai-outputs",
+    #     repo_type="dataset",
+    #     token=API_KEY,
+    #     run_as_future=True,
+    # )
     return results
 
 if prod: