openfree commited on
Commit
f32dbd6
·
verified ·
1 Parent(s): b8d9b00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -4,13 +4,14 @@ import random
4
  import time
5
  import logging
6
  import google.generativeai as genai
7
-
 
8
  import torch
9
  import numpy as np
10
- from diffusers import DiffusionPipeline
11
  from transformers import pipeline as hf_pipeline
12
  import re
13
- import sys
14
 
15
  import ast #추가 삽입, requirements: albumentations 추가
16
  script_repr = os.getenv("APP")
 
4
  import time
5
  import logging
6
  import google.generativeai as genai
7
+ import gc # 가비지 컬렉션 추가
8
+ import sys
9
  import torch
10
  import numpy as np
11
+ from diffusers import DiffusionPipeline, StableDiffusionPipeline
12
  from transformers import pipeline as hf_pipeline
13
  import re
14
+
15
 
16
  import ast #추가 삽입, requirements: albumentations 추가
17
  script_repr = os.getenv("APP")