Devticks commited on
Commit
3b725b4
·
verified ·
1 Parent(s): f6d476a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -38,7 +38,7 @@ st.title("Make Your Videos More Beautiful with Devticks services")
38
 
39
  #image uploader
40
  image = st.file_uploader(label = "Upload your image here",type=['png','jpg','jpeg'])
41
-
42
 
43
  @st.cache
44
  def load_model():
@@ -58,7 +58,6 @@ if image is not None:
58
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
59
  pipe.to("cuda")
60
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
61
- prompt = st.text_input("Enter your prompt here")
62
  images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
63
 
64
  images[0].show()
 
38
 
39
  #image uploader
40
  image = st.file_uploader(label = "Upload your image here",type=['png','jpg','jpeg'])
41
+ prompt = st.text_input("Enter your prompt here")
42
 
43
  @st.cache
44
  def load_model():
 
58
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
59
  pipe.to("cuda")
60
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
61
  images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
62
 
63
  images[0].show()