Update app.py

app.py

@@ -119,9 +119,21 @@ def generate(image, prompt_user, progress=gr.Progress(track_tqdm=True)):
         prompt = prompt_structure + prompt_user
         
         # Process input image
-        cropped_image = square_center_crop(image)
-        logo_dupli = duplicate_horizontally(cropped_image)
-        
+        try:
+            cropped_image = square_center_crop(image)
+        except Exception as e:
+            error_message = f"Error during cropping: {str(e)}"
+            print(error_message)  # For logging
+            raise gr.Error(error_message)
+
+        try:
+            logo_dupli = duplicate_horizontally(cropped_image)
+        except Exception as e:
+            error_message = f"Error during duplication: {str(e)}"
+            print(error_message)  # For logging
+            raise gr.Error(error_message)
+
+        print("just before getting into pipe")
         # Generate output
         out = pipe(
             prompt=prompt,