Update app.py

app.py

@@ -1,24 +1,28 @@
+import os
 import torch
 from diffusers import StableDiffusionPipeline
 import gradio as gr
 
-# Hugging Face Token (replace with your token)
-HF_TOKEN = "hf_your_token_here"
+# Hugging Face token from Secrets (for private model access)
+HF_TOKEN = os.getenv("HF_TOKEN")
 
-# Load gated model with auth token
+# Load the Anything-V3 model
 pipe = StableDiffusionPipeline.from_pretrained(
     "Linaqruf/anything-v3.0",
     use_auth_token=HF_TOKEN,
     torch_dtype=torch.float16
 ).to("cuda")
 
+# Main generation function
 def generate_thumbnail(prompt):
     image = pipe(prompt).images[0]
     return image
 
+# Gradio UI
 gr.Interface(
     fn=generate_thumbnail,
-    inputs=gr.Textbox(label="Prompt"),
+    inputs=gr.Textbox(label="Enter Thumbnail Prompt"),
     outputs=gr.Image(type="pil"),
-    title="Anime Thumbnail Generator"
+    title="Anime/Minecraft Style Thumbnail Generator",
+    description="Powered by Linaqruf/anything-v3.0 model"
 ).launch()