File size: 950 Bytes
7351cf1
 
 
6d5c0f5
 
88a060e
 
 
2981f10
88a060e
 
 
 
 
 
 
 
 
7107a1e
 
 
6d5c0f5
20ebd6e
6d5c0f5
 
 
 
 
 
 
 
 
 
20ebd6e
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import streamlit as st

x = st.slider('Select a value')
st.write(x, 'squared is', x * x)

from diffusers import DiffusionPipeline
import torch

pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32, use_safetensors=True, variant="fp32")


# if using torch < 2.0
# pipe.enable_xformers_memory_efficient_attention()

prompt = "An astronaut riding a green horse"

images = pipe(prompt=prompt).images[0]

st.image(images)

st.write("hello")

"""
import json
import requests
API_URL = "https://api-inference.huggingface.co/models/gpt2"
token = "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"
headers = {"Authorization": f"Bearer {token}"}
def query(payload):
    data = json.dumps(payload)
    response = requests.request("POST", API_URL, headers=headers, data=data)
    return json.loads(response.content.decode("utf-8"))
data = query("Can you please let us know more details about your ")
print(data)
"""