Update app.py

app.py

@@ -30,7 +30,7 @@ if authentication_status:
         st.text("上传彩色苹果叶子图片")
 
         pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)  
-        pipe = pipe.to("cuda")
+        pipe = pipe.to("cpu")#("cuda")
         
 
         uploaded_file = st.file_uploader("选择..", type=["jpg","png","jpeg"])
@@ -39,7 +39,8 @@ if authentication_status:
             st.image(image, caption='上传了图片。', use_column_width=True)
             st.write("")
             st.write("分类...")
-            generator = torch.Generator("cuda").manual_seed(1024)
+            # generator = torch.Generator("cuda").manual_seed(1024)
+            generator = torch.Generator().manual_seed(1024)
 
             image_gen = pipe(prompt, generator=generator).images[0]
             st.image(generator)