Update app.py

app.py

@@ -2,7 +2,7 @@ import diffusers
 import torch
 import os
 import time
-
+import accelerate
 import streamlit as st
 
 from stqdm import stqdm
@@ -59,6 +59,7 @@ def main():
     st.image(sample, caption="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(
@@ -69,7 +70,7 @@ def main():
     seed = st.slider("Seed for images", 1, 10000, 1)
 
     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 = []