Cognomen commited on
Commit
3cab9bb
Β·
1 Parent(s): b00ba88

copy first

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -32,7 +32,10 @@ generator = torch.manual_seed(0)
32
  def infer(prompt, negative_prompt, image):
33
  # implement your inference function here
34
 
35
- cond_input = conditioning_image_transforms(T.PILToTensor(image))
 
 
 
36
 
37
  output = pipe(
38
  prompt,
 
32
  def infer(prompt, negative_prompt, image):
33
  # implement your inference function here
34
 
35
+ inp = np.copy(image)
36
+ inp.setflags(writable=False)
37
+
38
+ cond_input = conditioning_image_transforms(image)
39
 
40
  output = pipe(
41
  prompt,