fix

app.py

@@ -40,10 +40,10 @@ if not model_file.exists():
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Using device: {device}")
 
-# Initialize configuration
+# Initialize configuration - use the correct tokenizer for SD 2.1
 config = Config(
     device=DeviceConfig(device=device),
-    tokenizer=CLIPTokenizer.from_pretrained("stabilityai/stable-diffusion-2-1")
+    tokenizer=CLIPTokenizer.from_pretrained("laion/CLIP-ViT-H-14-laion2B-s32B-b79K")
 )
 
 # Load models