Update app.py

app.py

@@ -43,13 +43,13 @@ def pred_func(image, prompt):
     if type(out) == type({}):
         #return out["spiga_seg"]
         control_image = out["spiga_seg"]
-        if type(control_image) == type("") and os.path.exists(control_image):
-            control_image = Image.open(control_image).convert("RGB")
+        if type(image) == type("") and os.path.exists(image):
+            image = Image.open(image).convert("RGB")
         elif hasattr(image, "shape"):
-            control_image = Image.fromarray(control_image).convert("RGB")
+            image = Image.fromarray(image).convert("RGB")
         else:
-            control_image = control_image.convert("RGB")
-        control_image = control_image.resize((512, 512))
+            image = image.convert("RGB")
+        image = image.resize((512, 512))
 
         generator = torch.manual_seed(0)
         image = pipe(