Update app.py

app.py

@@ -5,8 +5,10 @@ from PIL import Image
 import uuid
 import random
 
-proc1=gr.Interface.load("spaces/Omnibus/fast-diff")
-proc2=gr.Interface.load("spaces/Omnibus/fast-diff")
+proc1=diff
+proc2=diff 
+#proc1=gr.Interface.load("spaces/Omnibus/fast-diff")
+#proc2=gr.Interface.load("spaces/Omnibus/fast-diff")
 
 sky = "https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/sky.png"
 platform = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/platform.png'
@@ -41,10 +43,10 @@ def sky_fn(inp):
     #dif = models[0]
     uid=uuid.uuid4()
     output=proc1(inp,5,1)
-    in_url=f'https://omnibus-fast-diff.hf.space/file={output[0]}'
+    #in_url=f'https://omnibus-fast-diff.hf.space/file={output[0]}'
     #output=dif(inp)
     print(output)
-    outp=Image.open(in_url)
+    outp=Image.open(output[0])
     width, height = outp.size
     rat = width/height
     if width > height:
@@ -66,13 +68,13 @@ def star_fn(inp):
     #dif = gr.Interface.load("models/stabilityai/stable-diffusion-2-1-base", live=True),
     #dif = models[0]
     uid=uuid.uuid4()
-    output=proc2(inp,5,1)
-    #output=proc2.send_it(inp,5,1)
+    #output=proc2(inp,5,1)
+    output=proc2.send_it(inp,5,1)
     #output=dif(inp)
-    in_url=f'https://omnibus-fast-diff.hf.space/file={output[0]}'
+    #in_url=f'https://omnibus-fast-diff.hf.space/file={output[0]}'
     
     print(output)
-    outp=Image.open(in_url)
+    outp=Image.open(output[0])
     width, height = outp.size
     rat = width/height
     if width > height:
@@ -105,4 +107,4 @@ with gr.Blocks() as app:
     btn.click(sky_fn,prompt,[out_im_sky,out_sky_url])
     btn.click(star_fn,prompt,[out_im_star,out_star_url])
     app.load(sky_fn,start_prompt,[out_im_sky,out_sky_url]).then(game_fn,[out_sky_url],html_game)
-app.launch()
+app.queue(concurrency_count=10).launch()