Update app.py

app.py

@@ -1,28 +1,14 @@
-import os
 import torch
 import gradio as gr
 from diffusers import StableDiffusion3Pipeline
-from huggingface_hub import login
-
-# Load Hugging Face token from environment variable
-hf_token = os.getenv("HF_TOKEN")
-
-# Authenticate using the Hugging Face token
-if hf_token is not None:
-    login(token=hf_token)
-else:
-    raise ValueError("Please set your Hugging Face token as the HF_TOKEN environment variable.")
 
 def image_generator(prompt):
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    # Load the model using the authenticated token
-    pipeline = StableDiffusion3Pipeline.from_pretrained(
-        "stabilityai/stable-diffusion-3-medium-diffusers",
-        torch_dtype=torch.float16 if device == "cuda" else torch.float32,
-        text_encoder_3=None,
-        tokenizer_3=None,
-        use_auth_token=hf_token  # Use the token for authentication
-    )
+    pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", 
+                                                         torch_dtype=torch.float16 if device == "cuda" else torch.float32,
+                                                         text_encoder_3=None,
+                                                         tokenizer_3 = None)
+    #pipeline.enable_model_cpu_offload()
     pipeline.to(device)
 
     image = pipeline(
@@ -34,14 +20,19 @@ def image_generator(prompt):
         guidance_scale=9.0
     ).images[0]
 
+    #image.show()
     return image
 
+#image_generator("A magical cat doing spell")
+
 interface = gr.Interface(
-    fn=image_generator,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt..."),
-    outputs=gr.Image(type="pil"),
-    title="Image Generator App",
-    description="This is a simple image generator app using HuggingFace's Stable Diffusion 3 model."
-)
+    fn=image_generator, 
+    inputs=gr.Textbox(lines=2, placeholder = "Enter your prompt..."), 
+    outputs=gr.Image(type = "pil"),
+    title = "Image Generator App",
+    description = "This is a simple image generator app using HuggingFace's Stable Diffusion 3 model.")
 
 interface.launch()
+
+
+print(interface)