fix package reference

app.py

@@ -1,18 +1,29 @@
 import gradio as gr
 import os
 import torch
+if os.environ.get("SPACES_ZERO_GPU") is not None:
+    import spaces
+else:
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+
+            return wrapper
+
 from diffusers.utils import check_min_version
 from flux.controlnet_flux import FluxControlNetModel
 from flux.transformer_flux import FluxTransformer2DModel
 from flux.pipeline_flux_controlnet_inpaint import FluxControlNetInpaintingPipeline
-import huggingface_hub
+#import huggingface_hub
 
 # Import configuration
 import config
 
 # Login to Hugging Face Hub
-huggingface_hub.login(os.getenv('HF_TOKEN'))
-check_min_version("0.30.2")
+# huggingface_hub.login(os.getenv('HF_TOKEN_FLUX'))
+# check_min_version("0.30.2")
 
 # Load models
 transformer = FluxTransformer2DModel.from_pretrained(
@@ -28,7 +39,7 @@ pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     controlnet=controlnet,
     transformer=transformer,
     torch_dtype=torch.bfloat16
-).to("cuda")
+)#.to("cuda")
 pipe.transformer.to(torch.bfloat16)
 pipe.controlnet.to(torch.bfloat16)