Update app.py

app.py

@@ -35,15 +35,17 @@ if authentication_status:
 
         uploaded_file = st.file_uploader("选择..", type=["jpg","png","jpeg"])
         if uploaded_file is not None:
-            image = Image.open(uploaded_file).convert('RGB')
-            st.image(image, caption='上传了图片。', use_column_width=True)
-            st.write("")
-            st.write("分类...")
+            # image = Image.open(uploaded_file).convert('RGB')
+            # st.image(image, caption='上传了图片。', use_column_width=True)
+            # st.write("")
+            # st.write("分类...")
             # generator = torch.Generator("cuda").manual_seed(1024)
-            generator = torch.Generator().manual_seed(1024)
-
-            image_gen = pipe(prompt, generator=generator).images[0]
-            st.image(generator)
+            
+            # generator = torch.Generator().manual_seed(1024)
+            prompt = "a photograph of an astronaut riding a horse"
+            # image_gen = pipe(prompt, generator=generator).images[0]
+            image = pipe(prompt).images[0]
+            st.image(image)
 
 
         #     label = teachable_machine_classification(image, 'keras_model_apple.h5')