Remove upscling

app.py

@@ -594,21 +594,31 @@ def blend_details(input_image, relit_image, masked_image, scaling_factor=1):
 @spaces.GPU
 def generate_image(input_image_path, prompt):
 
-    resized_input_img = resize_to_square(input_image_path, 256)
+    # resized_input_img = resize_to_square(input_image_path, 256)
 
-    resized_input_img_path = '/tmp/gradio/resized_input_img.png'
+    # resized_input_img_path = '/tmp/gradio/resized_input_img.png'
 
-    resized_input_img.convert("RGBA").save(resized_input_img_path, "PNG")
+    # resized_input_img.convert("RGBA").save(resized_input_img_path, "PNG")
 
-    ai_gen_image = generate_ai_bg(resized_input_img, prompt)
+    # ai_gen_image = generate_ai_bg(resized_input_img, prompt)
+
+    # upscaled_ai_image = upscale_image(ai_gen_image, 8192)
+
+    # upscaled_input_image = upscale_image(resized_input_img, 8192)
+
+    # mask_input_image = run_grounded_sam(upscaled_input_image)
 
-    upscaled_ai_image = upscale_image(ai_gen_image, 8192)
+    # final_image = blend_details(upscaled_input_image, upscaled_ai_image, mask_input_image)
 
-    upscaled_input_image = upscale_image(resized_input_img, 8192)
+    # FAL
+
+    resized_input_img = resize_to_square(input_image_path, 1024)
+
+    ai_gen_image = generate_ai_bg(resized_input_img, prompt)
 
-    mask_input_image = run_grounded_sam(upscaled_input_image)
+    mask_input_image = run_grounded_sam(resized_input_img)
 
-    final_image = blend_details(upscaled_input_image, upscaled_ai_image, mask_input_image)
+    final_image = blend_details(resized_input_img, ai_gen_image, mask_input_image)
 
     return final_image