Update app.py

app.py

@@ -5,11 +5,13 @@ import numpy as np
 import PIL
 import cv2
 
+import PIL.Image
+import random
+
 print('\nDEBUG:   Version: 3')
 
 #pipeline = LDMTextToImagePipeline.from_pretrained("fusing/latent-diffusion-text2im-large")
-pipeline = LDMTextToImagePipeline.from_pretrained("CompVis/ldm-text2im-large-256")
-generator = torch.manual_seed(42)
+ldm_pipeline = LDMTextToImagePipeline.from_pretrained("CompVis/ldm-text2im-large-256")
 
 FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=milyiyo.testing-diffusers" />'
 
@@ -31,9 +33,15 @@ def genimage(prompt, iterations):
   #img = cv2.imread(file_name)
   ##cv2_imshow(img)
   #return img
+  
+def predict(prompt, steps=100):
+  torch.cuda.empty_cache()
+  generator = torch.manual_seed(42)
+  images = ldm_pipeline([prompt], generator=generator, num_inference_steps=steps, eta=0.3, guidance_scale=6.0)["sample"]
+  return images[0]
 
 iface = gr.Interface(
-  fn=genimage, 
+  fn=predict, 
   inputs=["text", "number"], 
   outputs=gr.Image(shape=[256,256], type="pil", elem_id="output_image"))
 iface.launch()