Update app.py

app.py

@@ -160,26 +160,27 @@ def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
 
     gallery_items = [(item[0], item[1]) for item in history]
 #    tables_html = "".join(f'<div style="margin-bottom:12px">{make_html_table(item[1])}</div>' for item in history)
-    del img
 
-    # Pythonのガベージコレクタを強制実行し、不要なメモリを回収します
-    import gc
-    gc.collect()
-        # GPUメモリを使用している場合、PyTorchのキャッシュをクリアします
-    if torch.cuda.is_available():
-        torch.cuda.empty_cache()
-    return img, gallery_items
+import gc
+import torch
+def process_image(img, gallery_items): # Assuming this is part of a function
+    try:
+        gc.collect()        
+        # Clear PyTorch's cache if GPU memory is being used
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+        return img, gallery_items
 
     except RuntimeError as e:
-        # CUDA Out of Memoryなどのエラーをキャッチ
-        error_message = f"画像生成中にエラーが発生しました: {e}\n\n"
-    if "CUDA out of memory" in str(e):
-        error_message += "GPUメモリが不足している可能性があります。画像のサイズを小さくするか、ステップ数を減らしてみてください。"
-    else:
-        error_message += "原因不明のエラーです。入力パラメータを確認してください。"
-    print(error_message) # サーバーログに出力
+        # Catch errors like CUDA Out of Memory
+        error_message = f"error in generate: {e}\n\n"
+        if "CUDA out of memory" in str(e):
+            error_message += "memory error"
+        else:
+            error_message += "other error"
+        print(error_message) # Output to server logs
+        return None, None
 
-    
 # CSS 設定（ダークモード強制防止＋カフェ風テーマ）
 css = """
 @import url('https://fonts.googleapis.com/css2?family=Playpen+Sans+Hebrew:wght@100;200;300;400;500;600;700;800&display=swap');