Update app.py

app.py

@@ -24,10 +24,12 @@ def save_image(img, prompt):
     unique_name = str(uuid.uuid4()) + ".png"
     img.save(unique_name)
 
-    # save with prompt to save prompt as image file name
+    # save with promp to save prompt as image file name
     filename = f"{prompt}.png"
     img.save(filename)
     return filename
+    
+    return unique_name
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
@@ -44,6 +46,7 @@ MAX_SEED = np.iinfo(np.int32).max
 USE_TORCH_COMPILE = 0
 ENABLE_CPU_OFFLOAD = 0
 
+
 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(
         "fluently/Fluently-XL-v4",
@@ -52,10 +55,13 @@ if torch.cuda.is_available():
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     
+    
     pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
     pipe.set_adapters("dalle")
 
     pipe.to("cuda")
+    
+    
 
 @spaces.GPU(enable_queue=True)
 def generate(
@@ -65,10 +71,15 @@ def generate(
     seed: int = 0,
     width: int = 1024,
     height: int = 1024,
+    #width: int = 1920,
+    #height: int = 1080,
     guidance_scale: float = 3,
+    #randomize_seed: bool = True,
     randomize_seed: bool = False,
     progress=gr.Progress(track_tqdm=True),
 ):
+
+    
     seed = int(randomize_seed_fn(seed, randomize_seed))
 
     if not use_negative_prompt:
@@ -81,25 +92,41 @@ def generate(
         height=height,
         guidance_scale=guidance_scale,
         num_inference_steps=20,
+        #num_inference_steps=50,
         num_images_per_prompt=1,
+        #cross_attention_kwargs={"scale": 2.00},
         cross_attention_kwargs={"scale": 0.65},
         output_type="pil",
     ).images
     image_paths = [save_image(img, prompt) for img in images]
+
+    #image_paths = [save_image(img) for img in images]
     download_links = [create_download_link(path) for path in image_paths]
 
     print(image_paths)
+    #return image_paths, seed
     return image_paths, seed, download_links
 
+examples_old = [
+"a modern hospital room with advanced medical equipment and a patient resting comfortably",
+"a team of surgeons performing a delicate operation using state-of-the-art surgical robots",
+"a elderly woman smiling while a nurse checks her vital signs using a holographic display",
+"a child receiving a painless vaccination from a friendly robot nurse in a colorful pediatric clinic",
+"a group of researchers working in a high-tech laboratory, developing new treatments for rare diseases",
+"a telemedicine consultation between a doctor and a patient, using virtual reality technology for a immersive experience"
+]
+
 examples = [
-    "An elderly man engages in a virtual reality physical therapy session, guided by a compassionate AI therapist that adapts the exercises to his abilities and provides encouragement, all from the comfort of his own home.",
-    "In a bright, welcoming dental office, a young patient watches in awe as a dental robot efficiently and painlessly repairs a cavity using a laser system, while the dentist explains the procedure using interactive 3D images.",
-    "A team of biomedical engineers collaborate in a state-of-the-art research facility, designing and testing advanced prosthetic limbs that seamlessly integrate with the patient's nervous system for natural, intuitive control.",
-    "A pregnant woman undergoes a routine check-up, as a gentle robotic ultrasound system captures high-resolution 3D images of her developing baby, while the obstetrician provides reassurance and guidance via a holographic display.",
-    "In a cutting-edge cancer treatment center, a patient undergoes a precision radiation therapy session, where an AI-guided system delivers highly targeted doses to destroy cancer cells while preserving healthy tissue.",
-    "A group of medical students attend a virtual reality lecture, where they can interact with detailed, 3D anatomical models and simulate complex surgical procedures under the guidance of renowned experts from around the world.",
-    "In a remote village, a local healthcare worker uses a portable, AI-powered diagnostic device to quickly and accurately assess a patient's symptoms, while seamlessly connecting with specialists in distant cities for expert advice and treatment planning.",
-    "At an advanced fertility clinic, a couple watches in wonder as an AI-assisted system carefully selects the most viable embryos for implantation, while providing personalized guidance and emotional support throughout the process."
+#"In a sleek, private hospital suite, a recovering patient relaxes in a luxurious bed while a personalized AI assistant monitors their health, adjusts the ambient lighting, and provides entertainment tailored to their preferences.",
+#"A skilled neurosurgeon leads a team of experts in a groundbreaking brain surgery, utilizing advanced neuronavigation systems and miniaturized robotics to precisely remove a tumor with minimal invasiveness.",
+"An elderly man engages in a virtual reality physical therapy session, guided by a compassionate AI therapist that adapts the exercises to his abilities and provides encouragement, all from the comfort of his own home.",
+"In a bright, welcoming dental office, a young patient watches in awe as a dental robot efficiently and painlessly repairs a cavity using a laser system, while the dentist explains the procedure using interactive 3D images.",
+"A team of biomedical engineers collaborate in a state-of-the-art research facility, designing and testing advanced prosthetic limbs that seamlessly integrate with the patient's nervous system for natural, intuitive control.",
+"A pregnant woman undergoes a routine check-up, as a gentle robotic ultrasound system captures high-resolution 3D images of her developing baby, while the obstetrician provides reassurance and guidance via a holographic display.",
+"In a cutting-edge cancer treatment center, a patient undergoes a precision radiation therapy session, where an AI-guided system delivers highly targeted doses to destroy cancer cells while preserving healthy tissue.",
+"A group of medical students attend a virtual reality lecture, where they can interact with detailed, 3D anatomical models and simulate complex surgical procedures under the guidance of renowned experts from around the world.",
+"In a remote village, a local healthcare worker uses a portable, AI-powered diagnostic device to quickly and accurately assess a patient's symptoms, while seamlessly connecting with specialists in distant cities for expert advice and treatment planning.",
+"At an advanced fertility clinic, a couple watches in wonder as an AI-assisted system carefully selects the most viable embryos for implantation, while providing personalized guidance and emotional support throughout the process."
 ]
 
 css = '''
@@ -110,6 +137,16 @@ visibility: hidden
 }
 '''
 
+
+#css = '''
+#.gradio-container{max-width: 560px !important}
+#h1{text-align:center}
+#footer {
+#    visibility: hidden
+#}
+#'''
+
+
 with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
     gr.Markdown(DESCRIPTION)
 
@@ -124,7 +161,6 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
             )
             run_button = gr.Button("Run", scale=0)
         result = gr.Gallery(label="Result", columns=1, preview=True, show_label=False)
-        gallery = gr.Gallery(label="Gallery", columns=2, rows=2, object_fit="cover", height="auto", show_label=False)
     with gr.Accordion("Advanced options", open=False):
         use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
         negative_prompt = gr.Text(
@@ -183,6 +219,7 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
         api_name=False,
     )
     
+
     gr.on(
         triggers=[
             prompt.submit,
@@ -200,7 +237,7 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
             guidance_scale,
             randomize_seed,
         ],
-        outputs=[result, seed, gallery],
+        outputs=[result, seed],
         api_name="run",
     )