Update app.py

app.py

@@ -1,28 +1,27 @@
-import os
-import torch
-from diffusers import StableDiffusionPipeline
-import gradio as gr
 
-# Hugging Face token from Secrets (for private model access)
-HF_TOKEN = os.getenv("HF_TOKEN")
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+import torch
 
-# Load the Anything-V3 model
+# Model load
 pipe = StableDiffusionPipeline.from_pretrained(
     "Linaqruf/anything-v3.0",
-    use_auth_token=HF_TOKEN,
-    torch_dtype=torch.float16
-).to("cuda")
+    torch_dtype=torch.float16,
+    revision="fp16",
+    safety_checker=None
+).to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Main generation function
 def generate_thumbnail(prompt):
-    image = pipe(prompt).images[0]
+    image = pipe(prompt, height=512, width=512).images[0]
     return image
 
 # Gradio UI
-gr.Interface(
+iface = gr.Interface(
     fn=generate_thumbnail,
-    inputs=gr.Textbox(label="Enter Thumbnail Prompt"),
-    outputs=gr.Image(type="pil"),
-    title="Anime/Minecraft Style Thumbnail Generator",
-    description="Powered by Linaqruf/anything-v3.0 model"
-).launch()
+    inputs=gr.Textbox(label="Thumbnail Prompt"),
+    outputs=gr.Image(label="Generated Thumbnail"),
+    title="Anime Thumbnail Generator",
+    description="Generate thumbnails using Linaqruf/anything-v3.0 model"
+)
+
+iface.launch()