Update app.py

app.py

@@ -111,7 +111,7 @@ def query(prompt, is_realistic, num_inference_steps, width, height, mod = None):
     payload = {
         "inputs": prompt,
         "seed": random.randint(1, 1000000000),
-        "parameters": {"num_inference_steps": num_inference_steps, "width": width, "height": height}
+        "parameters": {"num_inference_steps": num_inference_steps, "width": 1024, "height": 1024}
     }
 
     model = API_URL
@@ -131,18 +131,18 @@ def query(prompt, is_realistic, num_inference_steps, width, height, mod = None):
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
 
-        # image = resize_and_crop(image, 480, 640)
+        image = resize_and_crop(image, width, height)
 
         if (is_realistic == True):
-            image = add_noise(image, intensity = randint(100, 150) / 10)
-            image = compress_image(image, randint(75, 85))
+            image = add_noise(image, intensity = randint(25, 100) / 10)
+            image = compress_image(image, randint(80, 90))
 
             enhancer = ImageEnhance.Contrast(image)
-            image = enhancer.enhance(randint(75,85) / 100)
+            image = enhancer.enhance(randint(75,80) / 100)
             
             # Изменение насыщенности
             enhancer = ImageEnhance.Color(image)
-            image = enhancer.enhance(randint(70,80) / 100)
+            image = enhancer.enhance(randint(80,90) / 100)
     
             # Изменение экспозиции (яркости)
             enhancer = ImageEnhance.Brightness(image)