Update app.py

app.py

@@ -1,6 +1,8 @@
 import modal
 import gradio as gr 
 import numpy as np 
+from io import BytesIO
+import requests
 
 f = modal.Cls.lookup("casa-interior-hf-v3", "DesignModel")
 f_gc = modal.Cls.lookup("casa-interior-gc-v2", "GetProduct") 
@@ -59,7 +61,17 @@ def casa_ai_run_tab3(dict=None):
         print('Please provide cropped object')
         return None
     
-    result_image = f_gc.inference.remote(selected_crop)
+    result_urls = f_gc.inference.remote(selected_crop)
+    print(result_urls)
+
+    result_image = []
+    for r in result_urls: 
+        try: 
+            response = requests.get(r)
+            img = Image.open(BytesIO(response.content))
+            result_image.append(img)
+        except:
+            pass
     return result_image
 
 with gr.Blocks() as casa: