altanreis commited on
Commit
081826a
·
verified ·
1 Parent(s): 096de38

Create image_generater

Browse files
Files changed (1) hide show
  1. image_generater +15 -0
image_generater ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from diffusers import DiffusionPipeline
3
+ import torch
4
+ image_gen = DiffusionPipeline.from_pretrained(
5
+ "stabilityai/stable-diffusion-2",
6
+ torch_dtype=torch.float16,
7
+ use_safetensors=True,
8
+ variant="fp16"
9
+ ).to("cuda")
10
+
11
+ st.title("Metin Sınıflandırma Uygulaması")
12
+
13
+ user_input = st.text_area("Lütfen analiz edilecek metni girin:")
14
+ image = image_gen(prompt=user_input).images[0]
15
+ st.write(image)