FASTER! Change to inference API in app.py

app.py

@@ -1,21 +1,35 @@
+import io
+from os import getenv
 import gradio as gr
-from diffusers import DiffusionPipeline
-# model = "runwayml/stable-diffusion-v1-5"
-model = "ztjona/scopic-diffusion-OW-v1.4.1"
-generator = DiffusionPipeline.from_pretrained(model)
-prompt = "portrait photo of a old warrior chief"
+import requests
+from PIL import Image
+
+API_URL = (
+    "https://api-inference.huggingface.co/models/ztjona/scopic-diffusion-OW-v1.4.1"
+)
+API_TOKEN = getenv("API_TOKEN")
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
 
 def infer(prompt):
-    return generator(prompt).images[0]
+    payload = {"inputs": prompt}
+    response = requests.post(API_URL, headers=headers, json=payload)
+    image_bytes = response.content
 
+    return Image.open(io.BytesIO(image_bytes))
 
 
 demo = gr.Interface(
     fn=infer,
     inputs="text",
     outputs="image",
-    examples=[["city and clouds"]],
+    examples=[
+        ["city and clouds"],
+        ["tea party"],
+        ["mother working"],
+        ["buddhist monk"],
+        ["governor of old town"],
+    ],
     title="Generate images based on artist Oswaldo Guayasamín",
     description="Write a prompt and an image will be generated. ",
     article="""Fune tuned Stable Diffusion model. \n