Spaces:

Flo161
/

testdiffusion

Runtime error

Update app.py

7203681 about 3 years ago

306 Bytes

from diffusers import DiffusionPipeline
import torch
import streamlit as st

pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
pipeline.to("cuda")
image = pipeline("An image of a squirrel in Picasso style").images[0]

st.image(image, caption="Image")

	from diffusers import DiffusionPipeline
	import torch
	import streamlit as st

	pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
	pipeline.to("cuda")
	image = pipeline("An image of a squirrel in Picasso style").images[0]

	st.image(image, caption="Image")