Update app.py

app.py

@@ -1,5 +1,5 @@
 # from flask import Flask, request, jsonify
-# from handler import EndpointHandler
+from handler import EndpointHandler
 
 # app = Flask(__name__)
 # handler = EndpointHandler()
@@ -57,15 +57,18 @@ if image is not None:
     with st.spinner("🤖 AI is at Work! "):
         
 
-        result = reader.readtext(np.array(input_image))
+        # result = reader.readtext(np.array(input_image))
+        images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
+        images[0].show()
+        
 
-        result_text = [] #empty list for results
+        # result_text = [] #empty list for results
 
 
-        for text in result:
-            result_text.append(text[1])
+        # for text in result:
+        #     result_text.append(text[1])
 
-        st.write(result_text)
+        # st.write(result_text)
     #st.success("Here you go!")
     st.balloons()
 else: