change img

app.py

@@ -18,20 +18,21 @@ pipe.load_lora_weights(adapter_id)
 pipe.fuse_lora()
 
 def resize(w,h,img):
-    img = Image.open(img)
+    
     img = img.resize((w,h))
     return img
 
 def infer(source_img, prompt, steps, seed, Strength):
     start = time.time()
     print("开始")
+    img = Image.open(source_img)
     generator = torch.Generator(device).manual_seed(seed)  
     if int(steps * Strength) < 1:
         steps = math.ceil(1 / max(0.10, Strength))
-    w, h = source_img.size
+    w, h = img.size
     newW = 512
     newH = int(h * newW / w)
-    source_image = resize(newW,newH, source_img)
+    source_image = resize(newW,newH, img)
     source_image.save('source.png')
     image = pipe(prompt, image=source_image,width=newW,height=newH, strength=Strength, guidance_scale=0.0, num_inference_steps=steps).images[0]
     end = time.time()