Update app.py

app.py

@@ -1,18 +1,18 @@
+import os
 import gradio as gr
-from diffusers import StableDiffusionPipeline
-import torch
+from huggingface_hub import InferenceClient
 from PIL import Image, ImageDraw, ImageFont
 import io, zipfile
 import mediapipe as mp
 import numpy as np
 
 # ----------------------
-# Load Stable Diffusion
+# Hugging Face Inference Client
 # ----------------------
-pipe = StableDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-2",
-    torch_dtype=torch.float16,
-).to("cuda")  # Load on GPU
+client = InferenceClient(
+    provider="fal-ai",
+    api_key=os.environ.get("HF_TOKEN", "")
+)
 
 # Mediapipe for character detection
 mp_face = mp.solutions.face_detection.FaceDetection(model_selection=1, min_detection_confidence=0.5)
@@ -96,7 +96,7 @@ def add_speech_bubbles(img, bubbles, character_positions=None):
     return img
 
 def generate_comic(title, story, pages, panels, art_style, theme, dialogues, explicit):
-    """Generate AI comic panels with bubbles."""
+    """Generate AI comic panels with bubbles using Hugging Face Inference API."""
     comic_panels = []
     panels = int(panels)
     panel_dialogues = dialogues.split("\n") if dialogues else [""] * panels
@@ -104,9 +104,15 @@ def generate_comic(title, story, pages, panels, art_style, theme, dialogues, exp
     for i in range(panels):
         prompt = f"{story}, {art_style} comic style, {theme} theme, colorful, detailed, panel {i+1}"
         if explicit:
-            prompt += ", include blood, gore, offensive language, or dramatic action"
+            prompt += ", include blood, gore, offensive language, dramatic action"
+
+        # Generate image via HF API
+        image = client.text_to_image(
+            prompt,
+            model="stabilityai/stable-diffusion-3.5-large"
+        )
+        img = image.convert("RGB")
 
-        img = pipe(prompt, guidance_scale=7.5).images[0]
         character_positions = detect_characters(img)
 
         # Prepare bubbles