Update app.py

app.py

@@ -29,6 +29,14 @@ def update_game(sky=None,platform=None,star=None,bomb=None,dude=None):
 def dif_fn(inp):
     output=diff.send_it(inp,5,1)
     outp=Image.open(output[0])
+    width, height = outp.size
+    rat = width/height
+    if width > height:
+        outp = outp.resize(600*rat,600)
+    elif width < height:
+        outp = outp.resize(800,800*rat)     
+    else:
+        outp = outp.resize(800,800)
     outp.save("tmp_sky.png")
     out = os.path.abspath("tmp_sky.png")
     out_url = f'https://omnibus-game-test.hf.space/file={out}'