Duplicate from 3laa2/Text2img

Co-authored-by: Alaa Mahmoud <3laa2@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,14 @@
+---
+title: Text2img
+emoji: 🔥
+colorFrom: yellow
+colorTo: indigo
+sdk: streamlit
+sdk_version: 1.19.0
+app_file: app.py
+pinned: false
+license: openrail
+duplicated_from: 3laa2/Text2img
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,64 @@
+import streamlit as st
+import cv2 as cv
+import time
+import torch
+from diffusers import StableDiffusionPipeline
+
+
+def create_model(loc = "stabilityai/stable-diffusion-2-1-base", mch = 'cpu'):
+    pipe = StableDiffusionPipeline.from_pretrained(loc)
+    pipe = pipe.to(mch)
+    return pipe
+
+t2i = st.title("""
+Txt2Img
+###### `CLICK "Create_Update_Model"` :
+- `FIRST RUN OF THE CODE`
+- `CHANGING MODEL`""")
+
+the_type = st.selectbox("Model",("stabilityai/stable-diffusion-2-1-base",
+                                      "CompVis/stable-diffusion-v1-4"))
+
+create = st.button("Create The Model")
+
+if create:
+    st.session_state.t2m_mod = create_model(loc=the_type)
+
+prom = st.text_input("# Prompt",'')
+
+c1,c2,c3 = st.columns([1,1,3])
+c4,c5 = st.columns(2)
+
+with c1:
+  bu_1 = st.text_input("Seed",'999')
+with c2:
+  bu_2 = st.text_input("Steps",'12')
+with c3:
+  bu_3 = st.text_input("Number of Images",'1')
+with c4:
+  sl_1 = st.slider("Width",128,1024,512,8)
+with c5:
+  sl_2 = st.slider("hight",128,1024,512,8)
+
+st.session_state.generator = torch.Generator("cpu").manual_seed(int(bu_1))
+
+create = st.button("Imagine")
+
+if create:
+    model = st.session_state.t2m_mod
+    generator = st.session_state.generator
+
+    if int(bu_3) == 1 :
+      IMG = model(prom, width=int(sl_1), height=int(sl_2),
+                    num_inference_steps=int(bu_2),
+                    generator=generator).images[0]
+      st.image(IMG)
+        
+    else :
+      PROMS = [prom]*int(bu_3)
+        
+      IMGS = model(PROMS, width=int(sl_1), height=int(sl_2),
+                     num_inference_steps=int(bu_2),
+                     generator=generator).images
+    
+      st.image(IMGS)

requirements.txt

@@ -0,0 +1,10 @@
+streamlit
+altair<5
+opencv-python-headless
+numpy
+easyocr
+Pillow
+diffusers
+torch
+transformers
+accelerate