Update app.py

app.py

@@ -1,9 +1,9 @@
 from diffusers import AutoPipelineForText2Image, StableDiffusionImg2ImgPipeline
 from PIL import Image
 import gradio as gr
+import numpy as np
 import random
 import torch
-import numpy
 import math
 
 css = """
@@ -26,8 +26,9 @@ def generate(prompt, samp_steps, seed, strength, progress=gr.Progress(track_tqdm
 #        guidance_scale=0.0,
 #        generator=torch.manual_seed(seed),
 #    ).images[0]
-    imarray = numpy.random.rand(512,512,3) * 255
-    image = Image.fromarray(imarray.astype('uint8')).convert('RGBA')
+    data=np.random.randint(low=0,high=256,size=128*128*3, dtype=np.uint8)
+    data=data.reshape(128,128,3)
+    image = Image.fromarray(data,'RGB')
     upscaled_image = image.resize((1024,1024), 1)
     final_image = img2img(
         prompt,