yizhangliu commited on
Commit
a42aafd
1 Parent(s): 42193a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -39,8 +39,8 @@ import cv2
39
  import numpy as np
40
  import matplotlib.pyplot as plt
41
 
42
- ram_enable = False
43
- kosmos_enable = True
44
 
45
  lama_cleaner_enable = True
46
  try:
@@ -97,7 +97,7 @@ sam_predictor = None
97
  sam_mask_generator = None
98
  sd_model = None
99
  lama_cleaner_model= None
100
- lama_cleaner_model_device = 'cpu' # device
101
  ram_model = None
102
  kosmos_model = None
103
  kosmos_processor = None
 
39
  import numpy as np
40
  import matplotlib.pyplot as plt
41
 
42
+ ram_enable = True
43
+ kosmos_enable = False
44
 
45
  lama_cleaner_enable = True
46
  try:
 
97
  sam_mask_generator = None
98
  sd_model = None
99
  lama_cleaner_model= None
100
+ lama_cleaner_model_device = device
101
  ram_model = None
102
  kosmos_model = None
103
  kosmos_processor = None