sourceoftruthdata's picture
Update README.md
4615aaf
|
raw
history blame
924 Bytes
metadata
base_model: stabilityai/stable-diffusion-2-1
instance_prompt: a hand drawn painting in the style of picasso with geometric shapes
tags:
  - text-to-image
  - diffusers
  - autotrain
inference: true

DreamBooth trained by AutoTrain

Text encoder was not trained.
This is the model that feeds the Google Colab Notebook.
The model is a simplified version of the DreamBooth model.

Here is how to use the model:

import requests

API_URL = "https://api-inference.huggingface.co/models/sourceoftruthdata/sot_autotrain_dreambooth_v1" headers = {"Authorization": "Bearer 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"}

def query(payload): response = requests.post(API_URL, headers=headers, json=payload) return response.content image_bytes = query({ "inputs": "Astronaut riding a horse", })

You can access the image with PIL.Image for example

import io from PIL import Image image = Image.open(io.BytesIO(image_bytes))