Update app.py

app.py

@@ -31,12 +31,13 @@ def infer(prompt, init_image):
         text_embedding_optimization_steps=100,
         model_fine_tuning_optimization_steps=500)
     
-    with torch.no_grad():
-        torch.cuda.empty_cache()
+    #with torch.no_grad():
+    #    torch.cuda.empty_cache()
     
-    res = pipe(alpha=1)
+    #res = pipe(alpha=1)
 
-    return res.images[0]
+    #return res.images[0]
+    return 'trained success'
 
 title = """
     <div style="text-align: center; max-width: 650px; margin: 0 auto;">
@@ -108,12 +109,13 @@ with gr.Blocks(css=css) as block:
         prompt_input = gr.Textbox(label="Target text", placeholder="Describe the image with what you want to change about the subject")
         image_init = gr.Image(source="upload", type="filepath",label="Input Image")
         
-        submit_btn = gr.Button("Submit")
+        train_btn = gr.Button("Train")
         
         image_output = gr.Image(label="Edited image")
+        text_output = gr.Image(label="trained status")
         
         gr.HTML(article)
 
-    submit_btn.click(fn=infer, inputs=[prompt_input,image_init], outputs=[image_output])
+    submit_btn.click(fn=infer, inputs=[prompt_input,image_init], outputs=[text_output])
     
 block.queue(max_size=12).launch(show_api=False)