add login

app.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+from huggingface_hub import whoami
 import gradio as gr
 from diffusers import DiffusionPipeline
 
@@ -17,6 +19,7 @@ def generate_image_and_prompt(prompt: str):
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
+    gr.LoginButton()
     # Input Textbox for the user prompt
     input_prompt = gr.Textbox(label="Input Prompt", placeholder="Enter your prompt here")