1

app.py

@@ -32,12 +32,11 @@ class Pipe:
             h, w = image_tensor.shape[-2], image_tensor.shape[-1]
             h, w = h // 32 * 32, w // 32 * 32
             image_tensor = torch.nn.functional.interpolate(image_tensor, size=(h, w), mode='bilinear')
-            ipt=net_c(image_tensor)
+            ipt=self.net_c(image_tensor)
             image_tensor = image_tensor.half()
             ipt = ipt.half()
-            output_i, output_j=net_i(image_tensor,ipt,prompt=True)
-            output_j_out=[]
-            clean = output[-1][:, 3:, ...]
+            output_i, output_j=self.net_i(image_tensor,ipt,prompt=True)
+            clean = output_j[-1][:, 3:, ...]
             clean=torch.clamp(clean, 0, 1)
             self.output = clean