Update app.py

app.py

@@ -27,7 +27,7 @@ model_id = "CompVis/stable-diffusion-v1-4"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 context = autocast if device == "cuda" else nullcontext
 
-pipe = StableDiffusionPipeline.from_pretrained(model_id,use_auth_token=token).to(device)
+pipe = StableDiffusionPipeline.from_pretrained(model_id,revision="fp16", torch_dtype=torch.float16,use_auth_token=token).to(device)
 
 
 def infer(prompt,samples):
@@ -208,5 +208,6 @@ with demo:
     gr.Examples(examples=[["cont_img.png","ref_img.png"]],inputs=[overlay_img, ref_img], label = "Nvidia - FastPhotoStyle")
 
 
-demo.queue(concurrency_count=40, max_size=20).launch(enable_queue=True,max_threads=150)
+demo.queue(concurrency_count=40, max_size=20)
+demo.launch(share=True,enable_queue=True,max_threads=150)