Update app.py

app.py

@@ -16,6 +16,9 @@ from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_
 from huggingface_hub import HfFileSystem, ModelCard
 from safetensors.torch import load_file
 from huggingface_hub import login
+from gradio_client import Client 
+client = CLIENT("AlekseyCalvin/soonfactory6", hf_token=os.getenv("HF_TOKEN"))
+
 hf_token = os.environ.get("HF_TOKEN")
 login(token=hf_token)
 
@@ -80,7 +83,7 @@ def update_selection(evt: gr.SelectData, width, height):
         height,
     )
 
-@spaces.GPU(duration=70)
+@spaces.GPU()
 def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)