Update app.py

app.py

@@ -56,9 +56,13 @@ if image is not None:
 
     with st.spinner("🤖 AI is at Work! "):
         
-
-        # result = reader.readtext(np.array(input_image))
+        model_id = "timbrooks/instruct-pix2pix"
+        pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
+        pipe.to("cuda")
+        pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+        
         images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
+
         images[0].show()