Hugging Face's logo

Spaces:
m7mdal7aj
/
KB-VQA
Sleeping

App Files Community
2
m7mdal7aj commited on
Commit
46f4aef
·
1 Parent(s): c0bfbf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -3,7 +3,6 @@ import torch
3
 
4
  from transformers import Blip2Processor, Blip2ForConditionalGeneration
5
 
6
- global device = 'cuda' if torch.cuda.is_available() else 'cpu'
7
 
8
  def load_caption_model():
9
  # Quantization Config
 
3
 
4
  from transformers import Blip2Processor, Blip2ForConditionalGeneration
5
 
 
6
 
7
  def load_caption_model():
8
  # Quantization Config