Update app.py

app.py

@@ -39,30 +39,27 @@ qrcode_generator = qrcode.QRCode(
 # Define available models
 CONTROLNET_MODELS = {
     "QR Code Monster": "monster-labs/control_v1p_sd15_qrcode_monster",
-    "QR Code": "DionTimmer/controlnet_qrcode",
+    "QR Code": "DionTimmer/controlnet_qrcode-control_v1p_sd15",
     # Add more ControlNet models here
 }
 
 DIFFUSION_MODELS = {
     "Stable Diffusion v1.5": "ckpt/sd15",
-    "GhostMix v1.2": "digiplay/GhostMixV1.2VAE",
-    "Real-Dream": "luisrguerra/sd-1.5-real-dream",
+    "GhostMix": "sinkinai/GhostMix-V2-BakedVae",
+    "Real-Dream": "stablediffusionapi/edge-of-realism",
     # Add more diffusion models here
 }
 
 def load_models(controlnet_model, diffusion_model):
-    hf_token = get_hf_token()
     
     controlnet = ControlNetModel.from_pretrained(
         CONTROLNET_MODELS[controlnet_model],
-        use_auth_token=hf_token,
         torch_dtype=torch.float16
     ).to("cuda")
 
     pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
         DIFFUSION_MODELS[diffusion_model],
         controlnet=controlnet,
-        use_auth_token=hf_token,
         torch_dtype=torch.float16,
         safety_checker=None,
     ).to("cuda")
@@ -148,7 +145,7 @@ def invert_displayed_image(image):
         return Image.fromarray(inverted)
     return inverted
 
-@spaces.GPU()
+#@spaces.GPU()
 def inference(
     qr_code_content: str,
     prompt: str,
@@ -295,12 +292,13 @@ with gr.Blocks(theme='Hev832/Applio') as blocks:
             prompt = gr.Textbox(
                 label="Artistic Prompt",
                 placeholder="Describe the style or theme for your QR code art",
+                value="A high-resolution, photo-realistic minimalist rendering of Mount Fuji, depicted as a sharp, semi-realistic silhouette of a mountain range on the horizon. The mountain evokes strength and motion with clean, crisp lines and a sense of natural flow. The scene should feature detailed snow textures, subtle highlights on the mountain ridges, and a powerful yet serene atmosphere. The rendering should emphasize the strength of the mountain with a focus on clarity and precision in both texture and light. (Sharp outlines:1.5), (Photo-realistic:1.4), (Detailed textures:1.3), (Minimalist:1.3), (Semi-realistic:1.3), (Monochrome contrast:1.2), (Crisp detail:1.2), (Evoking strength:1.2), inspired by traditional Japanese woodblock prints, nature photography, and minimalist design principles.",
                 info="Be specific and creative! This guides the AI in creating your unique QR code art.",
             )
             negative_prompt = gr.Textbox(
                 label="Elements to Avoid",
                 placeholder="Describe what you don't want in the image",
-                value="ugly, disfigured, low quality, blurry, nsfw",
+                value="ugly, disfigured, low quality, blurry, nsfw, bad_pictures, (bad_prompt_version2:0.8), EasyNegative, 3d, cartoon, anime, sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), poorly drawn, distorted, overexposed, flat shading, bad proportions, deformed, pixelated, messy details, lack of contrast, unrealistic textures, bad anatomy, rough edges, low resolution, text artifacts.",
                 info="List elements or styles you want to avoid in your QR code art.",
             )
             use_qr_code_as_init_image = gr.Checkbox(
@@ -531,5 +529,5 @@ with gr.Blocks(theme='Hev832/Applio') as blocks:
         concurrency_limit=20
     )
 
-blocks.queue(max_size=20,api_open=False)
-blocks.launch(share=bool(os.environ.get("SHARE", False)), show_api=True)
+blocks.queue(max_size=20)
+blocks.launch(share=True, show_api=True)