Update app.py

app.py

@@ -259,7 +259,7 @@ import numpy as np
 from diffusers import FluxControlNetModel
 from diffusers.pipelines import FluxControlNetPipeline
 from PIL import Image
-from huggingface_hub import snapshot_download
+from huggingface_hub import snapshot_download,login
 import io
 import base64
 from flask import Flask, request, jsonify
@@ -283,6 +283,10 @@ else:
 
 # Load model from Huggingface Hub
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
+if huggingface_token:
+    login(token=huggingface_token)
+else:
+    print("Hugging Face token not found in environment variables.")
 print(huggingface_token)
 model_path = snapshot_download(
     repo_id="black-forest-labs/FLUX.1-dev",