Update app.py

app.py

@@ -24,12 +24,13 @@ TORCH_DTYPE = torch.float32 # float16/bfloat16 not practical on CPU
 MODEL_CONFIG = {
     "BlaireSilver13/youtube-thumbnail": {
         "repo_id": "BlaireSilver13/youtube-thumbnail",
-        "is_lora": False,
-        "base_model": None, # It's a full model
-        "pipeline_class": StableDiffusionPipeline,
-        "ip_adapter_repo": "h94/IP-Adapter", # Standard SD 1.5 IP-Adapter
+        "is_lora": True,
+        "lora_filename": "FLUX-youtube-thumbnails.safetensors", 
+        "base_model": "black-forest-labs/FLUX.1-dev",
+        "pipeline_class": AutoPipelineForText2Image, 
+        "ip_adapter_repo": "h94/IP-Adapter", 
         "ip_adapter_weights": "ip-adapter_sd15.bin",
-        "ip_adapter_image_encoder": "laion/CLIP-ViT-H-14-laion2B-s32B-b79K"
+        "ip_adapter_image_encoder": "laion/CLIP-ViT-H-14-laion2B-s32B-b79K" 
     },
     "itzzdeep/youtube-thumbnails-sdxl-lora": {
         "repo_id": "itzzdeep/youtube-thumbnails-sdxl-lora",