Update app.py

app.py

@@ -53,15 +53,19 @@ def main():
 
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
+    elapsed, pipeline = elapsed_time(load_pipeline_w_lora)
+    pipeline = pipeline.to(device)
+
+    st.write(f"Model is loaded in {elapsed} seconds!")
+    
     st.title("BAYC Text to IMAGE generator")
     st.write(f"Stable diffusion model is fine-tuned by lora using dataset {DATASET_REPO}")
 
     sample = Image.open(SAMPLE_IMAGE)
-    st.image(sample, caption="An ape with solid gold fur and beanie")
+    st.image(sample, caption="Example image with prompt <An ape with solid gold fur and beanie>")
 
-    elapsed, pipeline = elapsed_time(load_pipeline_w_lora)
-    pipeline = pipeline.to(device)
-    st.write(f"Model is loaded in {elapsed} seconds!")
+    
+    
 
     prompt = st.text_input(
         label="Write prompt to generate your unique BAYC image! (e.g. An ape with golden fur)")
@@ -71,7 +75,7 @@ def main():
     seed = st.text_input(label="Seed for images")
 
     if prompt and num_images and seed:
-        st.write(f"Generating {num_images} BAYC image with prompt {prompt}...")
+        st.write(f"Generating {num_images} BAYC image with prompt <{prompt}>...")
 
         generator = torch.Generator(device=device).manual_seed(seed)
         images = []
@@ -81,7 +85,7 @@ def main():
 
         st.write("Done!")
 
-        st.image(images, width=150, caption=f"Generated Images with {prompt}")
+        st.image(images, width=150, caption=f"Generated Images with <{prompt}>")
 
 if __name__ == '__main__':
     main()