5

app.py

@@ -39,7 +39,8 @@ class Pipe:
             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
+            
+            return clean
 
 pipe = Pipe()