Update app.py

app.py

@@ -11,16 +11,12 @@ import spaces
 from comfy import model_management
 
 CHROMA_VERSION = "chroma-unlocked-v48-detail-calibrated.safetensors"
-#LORA_VERSION = "DatasetTest4.safetensors"
+
 # Download required models
 t5_path = hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", filename="t5xxl_fp8_e4m3fn.safetensors", local_dir="models/text_encoders/")
 vae_path = hf_hub_download(repo_id="lodestones/Chroma", filename="ae.safetensors", local_dir="models/vae")
 unet_path = hf_hub_download(repo_id="lodestones/Chroma", filename=CHROMA_VERSION, local_dir="models/unet")
 
-#vae_path = hf_hub_download(repo_id="lodestones/Chroma", filename="ae.safetensors", local_dir="models/vae")
-#unet_path = hf_hub_download(repo_id="silveroxides/Chroma-Misc-Models", filename=CHROMA_VERSION, local_dir="models/unet")
-#lora_path = hf_hub_download(repo_id="Alme995/DatasetTest_Loras_Chroma", filename=LORA_VERSION, local_dir="models/loras")
-
 # Example prompts with their parameters
 EXAMPLES = [
     [
@@ -138,7 +134,7 @@ from nodes import (
     CLIPTextEncode,
     CLIPLoader,
     VAEDecode,
-    LoraLoaderModelOnly,
+    UNETLoader,
     VAELoader,
     SaveImage,
 )
@@ -160,20 +156,16 @@ saveimage = SaveImage()
 
 # Load models
 cliploader_78 = cliploader.load_clip(
-    #clip_name="flan-t5-xxl_float8_e4m3fn_scaled_stochastic.safetensors", type="chroma", device="default"
     clip_name="t5xxl_fp8_e4m3fn.safetensors", type="chroma", device="default"
 )
 t5tokenizeroptions_82 = t5tokenizeroptions.set_options(
-    min_padding=0,
-    min_length=0,
-    clip=get_value_at_index(cliploader_78, 0)
+    min_padding=1, min_length=0, clip=get_value_at_index(cliploader_78, 0)
 )
 unetloader_76 = unetloader.load_unet(
-    model_name=CHROMA_VERSION, chroma_hybrid_large=True, radiance_hybrid_large=False, chroma_hybrid_small=False, radiance_hybrid_small=False, wan=False, qwen=False
+    unet_name=CHROMA_VERSION, weight_dtype="fp8_e4m3fn"
 )
 vaeloader_80 = vaeloader.load_vae(vae_name="ae.safetensors")
 
-
 # Add all the models that load a safetensors file
 model_loaders = [cliploader_78, unetloader_76, vaeloader_80]
 
@@ -244,7 +236,6 @@ def generate_image(prompt, negative_prompt, width, height, steps, cfg, seed):
             images=get_value_at_index(vaedecode_79, 0),
         )
         
-        
         # Return the path to the saved image
         saved_path = f"output/{saved['ui']['images'][0]['filename']}"
         return saved_path
@@ -252,29 +243,25 @@ def generate_image(prompt, negative_prompt, width, height, steps, cfg, seed):
 # Create Gradio interface
 with gr.Blocks() as app:
     gr.Markdown(f"""
-# Chroma
-
-Model: [Chroma](https://huggingface.co/lodestones/Chroma) by [lodestones](https://huggingface.co/lodestones)
-
-Chroma1-HD-flash-heun : {CHROMA_VERSION}
-
-Run any ComfyUI Workflow on Spaces: [ComfyUI Workflows](https://huggingface.co/blog/run-comfyui-workflows-on-spaces)
-
-Space Author: [GitHub](https://github.com/gokayfem) | [X.com](https://x.com/gokayfem)
-    """)
+  # Chroma  
+  Model: [Chroma](https://huggingface.co/lodestones/Chroma) by [lodestones](https://huggingface.co/lodestones) 
+  Chroma Version: {CHROMA_VERSION}  
+  Run any ComfyUI Workflow on Spaces: [ComfyUI Workflows](https://huggingface.co/blog/run-comfyui-workflows-on-spaces)  
+  Compare Outputs: [DualView](https://dualview.ai) - Free side-by-side comparison tool  
+  Space Author: [GitHub](https://github.com/gokayfem) | [X.com](https://x.com/gokayfem)  
+      """)
     
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(
                 label="Prompt",
                 placeholder="Enter your prompt here...",
-                value="Overlaid at the center of the image is a title text that says \"CHROMA1-FLASH-HEUN\" in a large white 3D letters. This is a close-up photograph from a nature documentary capturing the right side of the face of a tiger. The photograph is centered on its highly detailed and speckled eye surrounded by intricately detailed fur. Amateur photography. unfiltered. natural lighting. anatomically correct. subtle shadows. perfect composition. highest quality. detailed. sharp focus",
                 lines=3
             )
             negative_prompt = gr.Textbox(
                 label="Negative Prompt",
                 placeholder="Enter negative prompt here...",
-                value="This greyscale unfinished sketch has bad proportions, is featureless and disfigured. It is a blurry ugly mess and with excessive gaussian blur. It is riddled with watermarks and signatures. Everything is smudged with leaking colors and nonsensical orientation of objects. Messy and abstract image filled with artifacts disrupt the coherency of the overall composition. The image has extreme chromatic abberations and inconsistent lighting. Dull, monochrome colors and countless artistic errors.",
+                value="low quality, ugly, unfinished, out of focus, deformed, disfigure, blurry, smudged, restricted palette, flat colors",
                 lines=2
             )
             
@@ -297,16 +284,16 @@ Space Author: [GitHub](https://github.com/gokayfem) | [X.com](https://x.com/goka
             with gr.Row():
                 steps = gr.Slider(
                     minimum=1,
-                    maximum=50,
-                    value=12,
+                    maximum=100,
+                    value=26,
                     step=1,
                     label="Steps"
                 )
                 cfg = gr.Slider(
                     minimum=1,
                     maximum=20,
-                    value=1,
-                    step=0.01,
+                    value=4,
+                    step=0.5,
                     label="CFG Scale"
                 )
                 seed = gr.Number(