Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import numpy as np
 import random
@@ -7,6 +8,7 @@ from diffusers import DiffusionPipeline
 import torch
 
 model_repo_id = "Qwen/Qwen-Image" 
+hf_token = os.getenv("HF_TOKEN")
 
 MODEL_CHOICES = [
     "black-forest-labs/FLUX.1-dev",
@@ -24,7 +26,7 @@ else:
     torch_dtype = torch.float32
     device = "cpu"
 
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype, use_auth_token=hf_token)
 pipe = pipe.to(device)
 
 current_model_id = model_repo_id