Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from PIL import Image
 import torch
 from diffusers import StableDiffusionPipeline
+import cv2 as cv
 
 st.title(" New App")
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16")
@@ -9,7 +10,7 @@ pipe.to("cpu")
 area = st.text_area("Area",'')
 if len(area)>5:
     image = pipe(area).images[0]
-    im = Image.open(image)
+    im = cv.imread(image)
     st.image(im)
 # else:
 #     st.text_input("NNNN",'OOOOO')