Update app.py

app.py

@@ -357,7 +357,7 @@ def inference(text):
         losses_str = ', '.join(f'{loss.item():g}' for loss in losses)
         tqdm.write(f'i: {i}, loss: {sum(losses).item():g}, losses: {losses_str}')
         out = synth(z)
-        TF.to_pil_image(out[0].cpu()).save('progress.png')
+        #TF.to_pil_image(out[0].cpu()).save('progress.png')
         #display.display(display.Image('progress.png'))
 
     def ascend_txt():
@@ -374,13 +374,13 @@ def inference(text):
             result.append(prompt(iii))
         img = np.array(out.mul(255).clamp(0, 255)[0].cpu().detach().numpy().astype(np.uint8))[:,:,:]
         img = np.transpose(img, (1, 2, 0))
-        imageio.imwrite('./steps/' + str(i) + '.png', np.array(img))
+        #imageio.imwrite('./steps/' + str(i) + '.png', np.array(img))
 
-        return result
+        return result, np.array(img)
 
     def train(i):
         opt.zero_grad()
-        lossAll = ascend_txt()
+        lossAll, image = ascend_txt()
         if i % args.display_freq == 0:
             checkin(i, lossAll)
         
@@ -389,19 +389,20 @@ def inference(text):
         opt.step()
         with torch.no_grad():
             z.copy_(z.maximum(z_min).minimum(z_max))
+        return image
 
     i = 0
     try:
         with tqdm() as pbar:
             while True:
-                train(i)
+                image = train(i)
                 if i == max_iterations:
                     break
                 i += 1
                 pbar.update()
     except KeyboardInterrupt:
         pass
-    return "./steps/300.png"
+    return image
 
 title = "VQGAN + CLIP"
 description = "Gradio demo for VQGAN + CLIP. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."
@@ -410,7 +411,7 @@ article = "<p style='text-align: center'>Originally made by Katherine Crowson (h
 gr.Interface(
     inference, 
     gr.inputs.Textbox(label="Input"), 
-    gr.outputs.Image(type="file", label="Output"),
+    gr.outputs.Image(type="numpy", label="Output"),
     title=title,
     description=description,
     article=article,