Update app.py

app.py

@@ -20,7 +20,7 @@ from diffusers import DiffusionPipeline, FluxPipeline
 import torch
 import gradio as gr
 
-# Corrected code with closed parenthesis
+# Corrected code with closed parenthesis and explicit token handling
 image_model = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", 
     torch_dtype=torch.bfloat16, 
@@ -190,4 +190,4 @@ class HuggingFaceChatbot:
 if __name__ == "__main__":
     logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
     chatbot = HuggingFaceChatbot()
-    chatbot.launch()
+    chatbot.launch()