Update app.py

app.py

@@ -1,12 +1,13 @@
 import os
 import random
-
+import requests
 import gradio as gr
 from gradio_client import Client, file
 import numpy as np
 from PIL import Image
 from typing import Tuple
 from translatepy import Translator
+from io import BytesIO
 
 MODEL = os.environ.get("MODEL")
 MODEL2 = os.environ.get("MODEL2")
@@ -108,16 +109,19 @@ def generate(
     		fn_index=1
         )
         print(result)
-        images = MODEL+"file="+result   
+        image_url = MODEL+"file="+result
+        response = requests.get(image_url)
+        images = Image.open(BytesIO(response.content))
     else:
         seed = 0
-        result = client2.predict(
+        job = client2.submit(
             prompt,
-            None,
+            "",
             7.5,
             "(No style)",
             fn_index=3    
         )
+        result = job.result()
         print(result)
         images = result["data"][0]