JoPmt commited on
Commit
0aebd0f
1 Parent(s): 772872b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from PIL import Image
2
  import gradio as gr
3
  import numpy as np
4
- import random, os
5
  import cv2
6
  import torch
7
  from accelerate import Accelerator
@@ -29,7 +29,7 @@ def make_hint(image, depth_estimator):
29
  return hint
30
 
31
  def plex(goof,prompt):
32
-
33
  goof = load_image(goof).resize((512,512))
34
  goof = goof.convert("RGB")
35
  goof.save('./gf.png', 'PNG')
 
1
  from PIL import Image
2
  import gradio as gr
3
  import numpy as np
4
+ import random, os, gc
5
  import cv2
6
  import torch
7
  from accelerate import Accelerator
 
29
  return hint
30
 
31
  def plex(goof,prompt):
32
+ gc.collect()
33
  goof = load_image(goof).resize((512,512))
34
  goof = goof.convert("RGB")
35
  goof.save('./gf.png', 'PNG')