Update app.py

app.py

@@ -75,7 +75,7 @@ def load_models_on_launch():
     loaded_controlnet = ControlNetModel.from_pretrained(
         controlnet_path,
         torch_dtype=torch.float16
-    ).to("mps")
+    ).to("cuda")
 
     diffusion_path = snapshot_download(DIFFUSION_MODELS["GhostMix"])
     loaded_pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
@@ -83,7 +83,7 @@ def load_models_on_launch():
         controlnet=loaded_controlnet,
         torch_dtype=torch.float16,
         safety_checker=None,
-    ).to("mps")
+    ).to("cuda")
     print("Models loaded successfully!")
 
 # Modify the load_models function to use global variables
@@ -234,7 +234,7 @@ def inference(
         if count_tokens(prompt) > MAX_TOKENS:
             raise gr.Error(f"Prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
 
-        if negative_prompt and count_tokens(negative_prompt) > MAX_TOKENS:
+        if count_tokens(negative_prompt) > MAX_TOKENS:
             raise gr.Error(f"Negative prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
 
         pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
@@ -243,7 +243,7 @@ def inference(
             seed = torch.seed()  # Generate a truly random seed
         generator = torch.manual_seed(seed)
 
-        if qr_code_content != "" or (qrcode_image is not None and qrcode_image.size == (1, 1)):
+        if qr_code_content != "" or qrcode_image.size == (1, 1):
             print("Generating QR Code from content")
             qr = qrcode.QRCode(
                 version=1,
@@ -256,11 +256,9 @@ def inference(
 
             qrcode_image = qr.make_image(fill_color=qr_color, back_color=bg_color)
             qrcode_image = resize_for_condition_image(qrcode_image, 1024)
-        elif qrcode_image is not None:
+        else:
             print("Using QR Code Image")
             qrcode_image = resize_for_condition_image(qrcode_image, 1024)
-        else:
-            raise gr.Error("No valid QR code content or image provided")
 
         # Determine which image to use as init_image and control_image
         if use_qr_code_as_init_image:
@@ -370,8 +368,8 @@ with gr.Blocks(theme='Hev832/Applio', css=css, fill_width=True, fill_height=True
             ![Yamamoto Logo](https://cdn-uploads.huggingface.co/production/uploads/64740cf7485q8e1bd51ac9/_VyYxp5qE_nRZ_LJqBxmL.webp)
             # 🎨 Yamamoto QR Code Art Generator
             ## Transform Your QR Codes into Brand Masterpieces
-            Welcome to Yamamoto's innovative QR Code Art Generator! This cutting-edge tool empowers our creative team to craft 
-            visually stunning, on-brand QR codes that perfectly blend functionality with artistic expression.
+            This cutting-edge tool empowers our creative team to craft visually stunning,<br> 
+            on-brand QR codes that perfectly blend functionality with artistic expression.
             ## 🚀 How It Works:
             1. **Enter Your QR Code Content**: Start by inputting the URL or text for your QR code.
             2. **Craft Your Prompt**: Describe the artistic style or theme you envision for your QR code.