Hugging Face's logo

Spaces:
MaykaGR
/
proyectomod
Sleeping

App Files Community
MaykaGR commited on
Commit
7a72478
路
verified 路
1 Parent(s): 9e0ec1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ import os
10
  os.environ['CUDA_VISIBLE_DEVICES'] = ''
11
 
12
  device = torch.device("cpu")
 
13
  model = model.to(device)
14
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
15
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large").to("cuda")
 
10
  os.environ['CUDA_VISIBLE_DEVICES'] = ''
11
 
12
  device = torch.device("cpu")
13
+ model = SomeModel()
14
  model = model.to(device)
15
  processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
16
  model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large").to("cuda")