nikigoli commited on
Commit
d3cb7f0
1 Parent(s): 0a55113

Reverted app to cpu

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,7 +43,7 @@ build_custom_prompter()
43
  from gradio_image_prompter import ImagePrompter
44
  #print("torch version")
45
  #print(torch.version.cuda)
46
- install_add_dependencies()
47
 
48
  class AppSteps(Enum):
49
  JUST_TEXT = 1
@@ -166,7 +166,7 @@ args = parser.parse_args()
166
  # args.device = torch.device('cuda')
167
  #else:
168
  # args.device = torch.device('cpu')
169
- args.device = torch.device('cuda')
170
  model, transform = build_model_and_transforms(args)
171
 
172
  examples = [
 
43
  from gradio_image_prompter import ImagePrompter
44
  #print("torch version")
45
  #print(torch.version.cuda)
46
+ #install_add_dependencies()
47
 
48
  class AppSteps(Enum):
49
  JUST_TEXT = 1
 
166
  # args.device = torch.device('cuda')
167
  #else:
168
  # args.device = torch.device('cpu')
169
+ args.device = torch.device('cpu')
170
  model, transform = build_model_and_transforms(args)
171
 
172
  examples = [