Update app_v4.py

app_v4.py

@@ -17,6 +17,7 @@ from huggingface_hub import CommitScheduler, HfApi, logging
 from debug import log_params, scheduler, save_image
 logging.set_verbosity_debug()
 from model_loader import safe_model_load
+from huggingface_hub.utils._runtime import dump_environment_info
 
 def hello(profile: gr.OAuthProfile | None) -> str:
     if profile is None:
@@ -55,8 +56,12 @@ pipe = FluxControlNetPipeline.from_pretrained(
 )
 pipe.to("cuda")
 
+try:
+    dump_environment_info()
+except Exception as e:
+    print(f"Failed to dump env info: {e}")
 
-@spaces.GPU(duration=12)
+@spaces.GPU(duration=6, progress=gr.Progress(track_tqdm=True))
 @torch.no_grad()
 def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale, seed, guidance_end):
     generator = torch.Generator().manual_seed(seed)
@@ -222,11 +227,13 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as demo:
         msg1 = gr.Markdown()
         try_btn = gr.LoginButton()
         try:
-            client = Client("LPX55/zerogpu-experiments", hf_token=huggingface_token)
+            x_ip_token = request.headers['x-ip-token']
+            client = Client("LPX55/zerogpu-experiments", hf_token=huggingface_token, headers={"x-ip-token": x_ip_token})
             cresult = client.predict(
             		n=3,
             		api_name="/predict"
             )
+            print(f"X TOKEN: {x_ip_token}")
             print(cresult)
         except:
             print("Guess we're just going to have to pretend that Spaces have been broken for almost a year now..")