Update app.py

app.py

@@ -1,30 +1,28 @@
-import gradio as gr
-from diffusers import StableDiffusionImg2ImgPipeline
 import torch
-from PIL import Image
+from diffusers import StableDiffusionPipeline
+import gradio as gr
 
-# Load pretrained model
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5",
-    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-    use_auth_token=True  # Use secret token
+# Model Load करो
+pipe = StableDiffusionPipeline.from_pretrained(
+    "Linaqruf/anything-v3.0",
+    torch_dtype=torch.float16,
+    revision="fp16",
+    safety_checker=None
 )
-pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+pipe = pipe.to("cuda")
 
-# Main function
-def generate_thumbnail(prompt, image):
-    image = image.resize((512, 512)).convert("RGB")
-    result = pipe(prompt=prompt, image=image, strength=0.75, guidance_scale=7.5)
-    return result.images[0]
+# Thumbnail generate करने वाला function
+def generate_thumbnail(prompt):
+    image = pipe(prompt).images[0]
+    return image
 
 # Gradio UI
-gr.Interface(
+interface = gr.Interface(
     fn=generate_thumbnail,
-    inputs=[
-        gr.Textbox(label="Prompt (e.g. 'Minecraft NOOB vs PRO battle')"),
-        gr.Image(type="pil", label="Upload Background Image")
-    ],
-    outputs=gr.Image(label="Generated Thumbnail"),
-    title="🖼️ AI Thumbnail Generator",
-    description="Enter prompt and image to create YouTube-style thumbnails"
-).launch()
+    inputs=gr.Textbox(label="Enter Prompt", placeholder="e.g. Minecraft thumbnail, anime character"),
+    outputs=gr.Image(type="pil", label="Generated Thumbnail"),
+    title="Thumbnail Generator (Anime Style)",
+    description="Enter your prompt to generate an anime-style thumbnail using Anything-v3.0"
+)
+
+interface.launch()