Test / app.py
altanreis's picture
Rename image_generater to app.py
61adf2f verified
raw
history blame
431 Bytes
import streamlit as st
from diffusers import DiffusionPipeline
import torch
image_gen = DiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-2",
torch_dtype=torch.float16,
use_safetensors=True,
variant="fp16"
).to("cuda")
st.title("Metin Sınıflandırma Uygulaması")
user_input = st.text_area("Lütfen analiz edilecek metni girin:")
image = image_gen(prompt=user_input).images[0]
st.write(image)