1

app.py

@@ -107,7 +107,7 @@ clip_model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-32', pr
 def run(prompt, steps, width, height, images, scale):
     opt = argparse.Namespace(
         prompt = prompt, 
-        outdir='./outputs',
+        ###outdir='./outputs',
         ddim_steps = int(steps),
         ddim_eta = 1,
         n_iter = 1,
@@ -124,15 +124,15 @@ def run(prompt, steps, width, height, images, scale):
     else:
         sampler = DDIMSampler(model)
     
-    os.makedirs(opt.outdir, exist_ok=True)
-    outpath = opt.outdir
+    ###os.makedirs(opt.outdir, exist_ok=True)
+    ###outpath = opt.outdir
 
     prompt = opt.prompt
 
 
-    sample_path = os.path.join(outpath, "samples")
-    os.makedirs(sample_path, exist_ok=True)
-    base_count = len(os.listdir(sample_path))
+    ###sample_path = os.path.join(outpath, "samples")
+    ###os.makedirs(sample_path, exist_ok=True)
+    ###base_count = len(os.listdir(sample_path))
 
     all_samples=list()
     all_samples_images=list()
@@ -171,7 +171,7 @@ def run(prompt, steps, width, height, images, scale):
                         else:
                             return(None,None,"Sorry, potential NSFW content was detected on your outputs by our NSFW detection model. Try again with different prompts. If you feel your prompt was not supposed to give NSFW outputs, this may be due to a bias in the model. Read more about biases in the Biases Acknowledgment section below.")
                         #Image.fromarray(x_sample.astype(np.uint8)).save(os.path.join(sample_path, f"{base_count:04}.png"))
-                        base_count += 1
+                        ###base_count += 1
                     all_samples.append(x_samples_ddim)
                     
     
@@ -182,7 +182,7 @@ def run(prompt, steps, width, height, images, scale):
     # to image
     grid = 255. * rearrange(grid, 'c h w -> h w c').cpu().numpy()
     
-    Image.fromarray(grid.astype(np.uint8)).save(os.path.join(outpath, f'{prompt.replace(" ", "-")}.png'))
+    ###Image.fromarray(grid.astype(np.uint8)).save(os.path.join(outpath, f'{prompt.replace(" ", "-")}.png'))
     #return(Image.fromarray(grid.astype(np.uint8)),all_samples_images,None)    
     return Image.fromarray(grid.astype(np.uint8))
         
@@ -221,7 +221,7 @@ app = gr.Interface(fn=engine,
                    #live=True,
                    description="Takes a text as input and reads it out to you.", 
                    outputs=[#gr.outputs.Textbox(type="auto", label="Text"),gr.outputs.Audio(type="file", label="Speech Answer"),
-                                      gr.outputs.Image(type="auto", label="output image"),],
+                                      gr.outputs.Image(type="auto", label="output image")],
                    examples=["On April 17th Sunday George celebrated Easter. He is staying at Empire State building with his parents. He is a citizen of Canada and speaks English and French fluently. His role model is former president Obama. He got 1000 dollar from his mother to visit Disney World and to buy new iPhone mobile.  George likes watching Game of Thrones."]
                    ).launch(debug=True)