Update app.py

app.py

@@ -42,7 +42,9 @@ def sanitize_filename(name):
     return re.sub(r"[^a-zA-Z0-9_-]", "_", name)[:80]
 
 @spaces.GPU(duration=75)
-def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+def infer(user_token, prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+    login(token=user_token)  # Authenticate for this call only
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
@@ -63,7 +65,6 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
         img_path = f"image_preview/{safe_name}_{seed}.jpg"
         img.convert("RGB").save(img_path, "JPEG", quality=60)
 
-        # Return image and updated list of previews
         previews = [f"image_preview/{f}" for f in sorted(os.listdir("image_preview")) if f.endswith(".jpg")]
         return img, seed, previews
 
@@ -86,6 +87,12 @@ Generate images in the surreal style of artist [Natalie Kav](https://www.behance
 """)
 
             with gr.Row():
+                hf_token_input = gr.Textbox(
+                    label="Your Hugging Face API Token",
+                    placeholder="Paste your token here",
+                    type="password",
+                )
+
                 prompt = gr.Text(
                     label="Prompt",
                     show_label=False,
@@ -109,14 +116,14 @@ Generate images in the surreal style of artist [Natalie Kav](https://www.behance
 
             result_example = gr.Image(visible=False)
 
-            gr.Examples(
-                examples=examples,
-                fn=infer,
-                inputs=[prompt],
-                outputs=[result_example, seed, gr.Gallery(visible=False)],
-                cache_examples=True,
-                cache_mode="lazy"
-            )
+                gr.Examples(
+                    examples=examples,
+                    fn=infer,
+                    inputs=[hf_token_input, prompt],
+                    outputs=[result_example, seed, gr.Gallery(visible=False)],
+                    cache_examples=False,  # Don't cache examples with real tokens!
+                )
+
 
         with gr.Column(scale=1, elem_id="right-column"):
             result = gr.Image(label="", show_label=False, elem_id="generated-image")
@@ -129,9 +136,10 @@ Generate images in the surreal style of artist [Natalie Kav](https://www.behance
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
-        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs=[hf_token_input, prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs=[result, seed, gallery],
     )
 
+
 if __name__ == "__main__":
     natalie_diffusion.launch()