Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 import gradio as gr
 import random
 import torch
+import numpy
 import math
 
 css = """
@@ -19,12 +20,14 @@ css = """
 def generate(prompt, samp_steps, seed, strength, progress=gr.Progress(track_tqdm=True)):
     if seed < 0:
         seed = random.randint(1,999999)
-    image = txt2img(
+    imagex = txt2img(
         prompt,
         num_inference_steps=1,
         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')
     upscaled_image = image.resize((1024,1024), 1)
     final_image = img2img(
         prompt,