Update app.py

app.py

@@ -1,9 +1,17 @@
+import os
 import gradio as gr
 from rembg import remove
 
+# Set environment variable for Numba
+os.environ['NUMBA_CACHE_DIR'] = '/tmp/numba_cache'
+
 def remove_background(input_image):
-    output = remove(input_image)
-    return output
+    try:
+        output = remove(input_image)
+        return output
+    except Exception as e:
+        print(f"Error processing image: {str(e)}")
+        return None
 
 interface = gr.Interface(
     fn=remove_background,