Update app.py

app.py

@@ -33,21 +33,17 @@ def infer(prompt, init_image):
         num_inference_steps=50,
         generator=generator,
         text_embedding_optimization_steps=500,
-        model_fine_tuning_optimization_steps=512)
+        model_fine_tuning_optimization_steps=600)
     
     with torch.no_grad():
         torch.cuda.empty_cache()
     
-    outputs = []
+    
     
     res = pipe(alpha=1)
-    outputs.append(res.images[0])
-    res = pipe(alpha=1.5)
-    outputs.append(res.images[0])
-    res = pipe(alpha=2)
-    outputs.append(res.images[0])
     
-    return outputs
+    
+    return res.images[0]
     #return 'trained success'
 
 title = """
@@ -122,11 +118,11 @@ with gr.Blocks(css=css) as block:
         
         submit_btn = gr.Button("Train")
         
-        gallery_output = gr.Gallery(label="Edited images")
+        image_output = gr.Gallery(label="Edited image")
         #text_output = gr.Image(label="trained status")
         
         gr.HTML(article)
 
-    submit_btn.click(fn=infer, inputs=[prompt_input,image_init], outputs=[gallery_output])
+    submit_btn.click(fn=infer, inputs=[prompt_input,image_init], outputs=[image_output])
     
 block.queue(max_size=12).launch(show_api=False)