Update app.py

app.py

@@ -1,37 +1,43 @@
 import gradio as gr
-import torch
-from transformers import DalleBartProcessor, DalleBartForConditionalGeneration
 from PIL import Image
 import numpy as np
 import cv2
+import requests
 
-# Load model and processor
-processor = DalleBartProcessor.from_pretrained("dalle-mini/dalle-mini/mega-1-fp16")
-model = DalleBartForConditionalGeneration.from_pretrained("dalle-mini/dalle-mini/mega-1-fp16")
+# Use the Hugging Face inference API for DALL·E Mini (public access, no token required)
+API_URL = "https://api-inference.huggingface.co/models/dalle-mini/dalle-mini/mega-1-fp16"
 
-# Force CPU
-model.to("cpu")
-
-def image_to_sketch(pil_img):
-    img = np.array(pil_img.convert("L"))
-    inv = 255 - img
+def image_to_sketch(image):
+    image = image.convert("L")  # convert to grayscale
+    inv = 255 - np.array(image)
     blur = cv2.GaussianBlur(inv, (21, 21), 0)
-    sketch = cv2.divide(img, 255 - blur, scale=256)
+    sketch = cv2.divide(np.array(image), 255 - blur, scale=256)
     return Image.fromarray(sketch)
 
 def generate_sketch(prompt):
-    prompt = prompt + ", pencil sketch, line drawing, black and white, minimal"
-    inputs = processor([prompt], return_tensors="pt").to("cpu")
+    full_prompt = prompt + ", pencil sketch, line art, black and white, minimal"
+    response = requests.post(API_URL, json={"inputs": full_prompt})
+    
+    if response.status_code != 200:
+        return f"Error generating image: {response.status_code}"
+
+    output = response.json()
+    if isinstance(output, dict) and output.get("error"):
+        return f"API Error: {output['error']}"
 
-    output_ids = model.generate(**inputs, max_length=256)
-    image = processor.decode(output_ids[0], output_type=Image)
+    # Grab image URL
+    image_url = output[0]['generated_image']
+    image_response = requests.get(image_url)
+    image = Image.open(BytesIO(image_response.content))
 
+    # Convert to sketch
     sketch = image_to_sketch(image)
     return sketch
 
 gr.Interface(
     fn=generate_sketch,
-    inputs=gr.Textbox(placeholder="Enter a description, e.g. 'a cat on a skateboard'"),
+    inputs=gr.Textbox(placeholder="Enter a description like 'a robot on a bike'"),
     outputs="image",
-    title="Text to Sketch Generator"
+    title="Text to Sketch AI",
+    description="This app turns your text prompts into pencil sketch-style images using DALL·E Mini + sketch filter."
 ).launch()