Update app.py

app.py

@@ -99,6 +99,7 @@ def generate(
 
     prompt = apply_style(style, prompt)
     strength = 0.7
+    images = []
     if use_turbo:
         result = client.predict(
             None,
@@ -111,7 +112,8 @@ def generate(
         print(result)
         image_url = MODEL+"file="+result
         response = requests.get(image_url)
-        images = Image.open(BytesIO(response.content))
+        response = Image.open(BytesIO(response.content))
+        images = images.append([(response)])
     else:
         seed = 0
         job = client2.submit(
@@ -123,7 +125,7 @@ def generate(
         )
         result = job.result()
         print(result)
-        images = result["data"][0]
+        images = result
 
     return images, seed