Update app.py

app.py

@@ -54,7 +54,7 @@ def main():
 
     
 
-    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}")
@@ -62,6 +62,8 @@ def main():
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
     elapsed, pipeline = elapsed_time(load_pipeline_w_lora)
+    st.write(f"Model is loaded in {elapsed} seconds!")
+    
     pipeline = pipeline.to(device)
 
     sample = Image.open(SAMPLE_IMAGE)