Update app.py

app.py

@@ -245,7 +245,9 @@ def inference(task, language_instruction, phrase_list, location_list, inpainting
     with torch.autocast(device_type='cuda', dtype=torch.float16):
         if task == 'User provide boxes' or 'Available boxes':
             if style_image == None:
-                return grounded_generation_box(get_model('base'), instruction, *args, **kwargs)
+                result = grounded_generation_box(get_model('base'), instruction, *args, **kwargs)
+                torch.cuda.empty_cache()
+                return result
             else:
                 return grounded_generation_box(get_model('style'), instruction, *args, **kwargs)