Update app.py

app.py

@@ -3,13 +3,15 @@ import diff
 import os
 from PIL import Image
 import uuid
-dif = gr.Interface.load("models/stabilityai/stable-diffusion-2-1-base", live=True),
-print (dif)
 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'
 star = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/star.png'
 bomb = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/bomb.png'
 dude = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/dude.png'
+models=[
+    gr.Interface.load("models/stabilityai/stable-diffusion-2-1-base",live=True)
+]
+
 
 def game_fn(sky=sky,platform=platform,star=star,bomb=bomb,dude=dude):
     html_mod=f"""
@@ -29,7 +31,8 @@ def update_game():
     return game_fn(sky=sky)
 
 def dif_fn(inp):
-    
+    #dif = gr.Interface.load("models/stabilityai/stable-diffusion-2-1-base", live=True),
+    dif = models[0]
     uid=uuid.uuid4()
     #output=diff.send_it(inp,5,1)
     output=dif(inp)