5

app.py

@@ -40,14 +40,15 @@ class Pipe:
             clean = output_j[-1][:, 3:, ...]
             clean=torch.clamp(clean, 0, 1)
             
-            return clean
+            self.output = clean
 
 pipe = Pipe()
 
 
 @spaces.GPU(duration=120)
 def predict(img):
-    return pipe(img)
+    pipe(img)
+    return pipe.output
 demo=gr.Interface(predict, gr.Image(), "image")
 
 demo.launch()