Update app.py

app.py

@@ -299,8 +299,14 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
         return None, None
 
         return generated_image
+    except Exception as e:
+        print(f"Error in image generation: {str(e)}")
+        return None
     finally:
-        clear_gpu_memory()  # 함수 종료 시 메모리 정리
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+            torch.cuda.synchronize()
+        gc.collect()
 
 def clear_gpu_memory():
     """GPU 메모리를 더 철저하게 정리하는 함수"""
@@ -414,9 +420,15 @@ def text_to_image(prompt: str, height: int, width: int, steps: int, scales: floa
         gc.collect()
 
         return generated_image
+    except Exception as e:
+        print(f"Error in image generation: {str(e)}")
+        return None
     finally:
-        clear_gpu_memory()  # 함수 종료 시 메모리 정리
-        
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+            torch.cuda.synchronize()
+        gc.collect()
+    
 css = """
 footer {
     visibility: hidden;