Update style_transfer.py

style_transfer.py

@@ -41,11 +41,11 @@ class DogStyleTransfer:
         # Define style to model mapping based on availability
         if self.device == "cuda":
             self.style_model_mapping = {
-                "Japanese Anime Style": "Linaqruf/anything-v3.0",  # Specialized for anime style
-                "Classic Cartoon": "nitrosocke/mo-di-diffusion",   # Specialized for Disney style
-                "Oil Painting": "runwayml/stable-diffusion-v1-5",  # Original model good for painting styles
-                "Watercolor": "dreamlike-art/dreamlike-photoreal-2.0",  # Photorealistic art style model
-                "Cyberpunk": "dreamlike-art/dreamlike-diffusion-1.0"  # Dreamlike style model
+                "Japanese Anime Style": "Linaqruf/anything-v3.0",  
+                "Classic Cartoon": "nitrosocke/mo-di-diffusion",   
+                "Oil Painting": "runwayml/stable-diffusion-v1-5",  
+                "Watercolor": "dreamlike-art/dreamlike-photoreal-2.0",  
+                "Cyberpunk": "dreamlike-art/dreamlike-diffusion-1.0"  
             }
         else:
             # Lightweight models for CPU mode
@@ -434,7 +434,7 @@ def create_style_transfer_tab(dog_style_transfer):
                 <p>After uploading your dog photo, the system will transform it into your chosen artistic style. Try different styles to create stunning effects!</p>
                 <p>The system uses specialized models for each style to ensure the best results.</p>
                 <p style="margin-top: 10px; padding: 8px; background-color: #fff9e6; border-left: 4px solid #ffd966; border-radius: 4px;"><b>⏱️ Patience is a virtue!</b> While AI is working its magic, your dog might have time to learn a new trick or two. The transformation can take up to 30 seconds, depending on how photogenic your furry friend is! 🐾</p>
-                <p style="margin-top: 10px; padding: 8px; background-color: #e6f9ff; border-left: 4px solid #66c2ff; border-radius: 4px;"><b>🤫 A Little Secret:</b> Although we designed this tool for dogs, it can actually transform any photo! Portraits, landscapes, even your favorite teddy bear — feel free to try them all! Just don't tell the other dogs… they might get jealous! 😉</p>
+                <p style="margin-top: 10px; padding: 8px; background-color: #e6f9ff; border-left: 4px solid #66c2ff; border-radius: 4px;"><b>🤫 A Little Secret:</b> Although I designed this tool for dogs, it can actually transform any photo! Portraits, landscapes, even your favorite teddy bear, feel free to try them all! Just don't tell the other dogs… they might get jealous! 😉</p>
                 <p style="margin-top: 10px; padding: 8px; background-color: #e6f9e6; border-left: 4px solid #66cc77; border-radius: 4px;"><b>✨ Unlimited Creativity!</b> Sometimes, AI might surprise you with unexpected creative interpretations, adding unique colors or features to your image. ✨</p>
             </div>
             """)
@@ -773,4 +773,4 @@ def create_style_transfer_tab(dog_style_transfer):
             </div>
             """)
 
-    return input_image, style_dropdown, style_button, output_image
+    return input_image, style_dropdown, style_button, output_image