Create main.py

main.py

@@ -0,0 +1,22 @@
+from fastapi import FastAPI
+from diffusers import DiffusionPipeline
+
+import json
+import requests
+
+from PIL import Image
+from io import BytesIO
+
+
+app = FastAPI()
+
+pipeline = DiffusionPipeline.from_pretrained("ogkalu/Comic-Diffusion")
+pipeline = pipeline.to("cpu")
+
+@app.get("/")
+def read_root():
+    prompt = "anime boy"
+    image = pipeline(prompt,width=256,height=256,num_inference_steps=10).images[0]
+    print('Success Ok')
+    return image
+