Update app.py

app.py

@@ -219,6 +219,9 @@ def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
     filepath = f"temp_{filename}" # ここで一時ファイルのパスを定義
 
     try:
+        gc.collect() # 追加
+        if torch.cuda.is_available(): # 追加
+            torch.cuda.empty_cache() # 追加
         
         if rand:
             seed = random.randint(0, MAX_SEED)
@@ -302,9 +305,9 @@ def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
         gallery_items = [(item[2], item[1].replace("|-|", "\n")) for item in history]
     
         processed_img, processed_gallery_items = process_image(img, gallery_items)
-                if processed_img is None or processed_gallery_items is None:
-                print("Image processing failed, skipping history update.")
-                return None, history_gallery, history_tables
+        if processed_img is None or processed_gallery_items is None:
+        print("Image processing failed, skipping history update.")
+        return None, history_gallery, history_tables
         latest_caption_table = make_html_table(caption_text)
         return processed_img, processed_gallery_items, latest_caption_table